diff --git a/src/TensorFlowNET.Core/Sessions/Session.cs b/src/TensorFlowNET.Core/Sessions/Session.cs
index 1e94b882..5b842408 100644
--- a/src/TensorFlowNET.Core/Sessions/Session.cs
+++ b/src/TensorFlowNET.Core/Sessions/Session.cs
@@ -39,7 +39,7 @@ namespace Tensorflow
 
         public static Session LoadFromSavedModel(string path)
         {
-            using var graph = new Graph();
+            var graph = new Graph();
             using var status = new Status();
             using var opt = c_api.TF_NewSessionOptions();
 
diff --git a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
index 5cdb28f7..975968cb 100644
--- a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
+++ b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
@@ -6,6 +6,7 @@ using System.Linq;
 using System.Reflection;
 using System.Text;
 using System.Threading.Tasks;
+using Tensorflow.NumPy;
 using static Tensorflow.Binding;
 
 namespace Tensorflow.Benchmark.Leak
@@ -21,7 +22,15 @@ namespace Tensorflow.Benchmark.Leak
 
 			for (var i = 0; i < 1024; i++)
             {
-				using var sess = Session.LoadFromSavedModel(ClassifierModelPath);
+				using (var sess = Session.LoadFromSavedModel(ClassifierModelPath)) {
+					using (var g = sess.graph.as_default()) {
+						var inputOp = g.OperationByName("inference_input");
+						var outputOp = g.OperationByName("StatefulPartitionedCall");
+
+						var inp = np.zeros(new Shape(new int[] { 1, 96, 2 }));
+						var ops = g.OperationByName("StatefulPartitionedCall");
+					}
+				}
 			}
 		}
 	}
diff --git a/src/TensorFlowNet.Benchmarks/Program.cs b/src/TensorFlowNet.Benchmarks/Program.cs
index 22abf730..e71b7514 100644
--- a/src/TensorFlowNet.Benchmarks/Program.cs
+++ b/src/TensorFlowNet.Benchmarks/Program.cs
@@ -14,8 +14,8 @@ namespace TensorFlowBenchmark
         {
             print(tf.VERSION);
 
-            /*new SavedModelCleanup().Run();
-            new RepeatDataSetCrash().Run();
+            new SavedModelCleanup().Run();
+            /*new RepeatDataSetCrash().Run();
             new GpuLeakByCNN().Run();*/
 
             if (args?.Length > 0)