diff --git a/tensorflow/go/graph.go b/tensorflow/go/graph.go
index 473175c6440..460c61098be 100644
--- a/tensorflow/go/graph.go
+++ b/tensorflow/go/graph.go
@@ -515,7 +515,7 @@ func LoadLibrary(path string) (*LibraryHandler, error) {
 		cptr: cptr,
 	}
 
-	runtime.SetFinalizer(h, (*LibraryHandler).free)
+	runtime.SetFinalizer(lh, (*LibraryHandler).free)
 	return lh, nil
 }