X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fldk%2Fstructs%2FLockedNetworkGraph.java;h=62647c2650be9f925647187e5bad047069b42d37;hb=fed2245c60159f6c074c9ed5c0f3ce273ad9841b;hp=614a5d969cb9db9b4675181edbe110b9d0a76ae5;hpb=ad15b3a4dbf3fbc7f08ef22d656bae79e8182008;p=ldk-java

diff --git a/src/main/java/org/ldk/structs/LockedNetworkGraph.java b/src/main/java/org/ldk/structs/LockedNetworkGraph.java
index 614a5d96..62647c26 100644
--- a/src/main/java/org/ldk/structs/LockedNetworkGraph.java
+++ b/src/main/java/org/ldk/structs/LockedNetworkGraph.java
@@ -2,16 +2,21 @@ package org.ldk.structs;
 
 import org.ldk.impl.bindings;
 import org.ldk.enums.*;
+import org.ldk.util.*;
+import java.util.Arrays;
 
+@SuppressWarnings("unchecked") // We correctly assign various generic arrays
 public class LockedNetworkGraph extends CommonBase implements AutoCloseable {
 	LockedNetworkGraph(Object _dummy, long ptr) { super(ptr); }
 	@Override public void close() {
-		bindings.LockedNetworkGraph_free(ptr);
+		if (ptr != 0) { bindings.LockedNetworkGraph_free(ptr); }
 	}
 
 	public NetworkGraph graph() {
-		NetworkGraph ret = new NetworkGraph(null, bindings.LockedNetworkGraph_graph(this.ptr));
-		return ret;
+		long ret = bindings.LockedNetworkGraph_graph(this.ptr);
+		NetworkGraph ret_hu_conv = new NetworkGraph(null, ret);
+		ret_hu_conv.ptrs_to.add(this);
+		return ret_hu_conv;
 	}
 
 }