diff --git a/src/TensorFlowNET.Core/Training/Saving/SavedModel/function_deserialization.cs b/src/TensorFlowNET.Core/Training/Saving/SavedModel/function_deserialization.cs
index af9fbeda..77b115a4 100644
--- a/src/TensorFlowNET.Core/Training/Saving/SavedModel/function_deserialization.cs
+++ b/src/TensorFlowNET.Core/Training/Saving/SavedModel/function_deserialization.cs
@@ -116,12 +116,8 @@ namespace Tensorflow.Training.Saving.SavedModel
             }
 
             Dictionary<string, ConcreteFunction> loaded_gradients = new();
-            // Debug(Rinne)
-            var temp = _sort_function_defs(library, function_deps);
-            int i = 0;
-            foreach (var fdef in temp)
+            foreach (var fdef in _sort_function_defs(library, function_deps))
             {
-                i++;
                 var orig_name = _fix_fdef_in_place(fdef, functions, load_shared_name_suffix, new_gradient_op_types);
 
                 object structured_input_signature = null;
diff --git a/src/TensorFlowNET.Core/Training/Saving/SavedModel/loader.cs b/src/TensorFlowNET.Core/Training/Saving/SavedModel/loader.cs
index ae7e2cf5..727d18a8 100644
--- a/src/TensorFlowNET.Core/Training/Saving/SavedModel/loader.cs
+++ b/src/TensorFlowNET.Core/Training/Saving/SavedModel/loader.cs
@@ -214,12 +214,6 @@ namespace Tensorflow
                     continue;
                 }
                 var proto = _proto.Nodes[node_id];
-                if(node_id == 10522)
-                {
-                    // Debug(Rinne)
-                    Console.WriteLine();
-                }
-                var temp = _get_node_dependencies(proto);
                 foreach (var dep in _get_node_dependencies(proto).Values.Distinct())
                 {
                     deps.Add(dep);
diff --git a/test/TensorFlowNET.Keras.UnitTest/Layers/ModelSaveTest.cs b/test/TensorFlowNET.Keras.UnitTest/Layers/ModelSaveTest.cs
index 647b2ad7..90f5f380 100644
--- a/test/TensorFlowNET.Keras.UnitTest/Layers/ModelSaveTest.cs
+++ b/test/TensorFlowNET.Keras.UnitTest/Layers/ModelSaveTest.cs
@@ -18,8 +18,8 @@ namespace TensorFlowNET.Keras.UnitTest
         {
             var model = GetFunctionalModel();
             var config = model.get_config();
-            Debug.Assert(config is ModelConfig);
-            var new_model = new ModelsApi().from_config(config as ModelConfig);
+            Debug.Assert(config is FunctionalConfig);
+            var new_model = new ModelsApi().from_config(config as FunctionalConfig);
             Assert.AreEqual(model.Layers.Count, new_model.Layers.Count);
         }