diff --git a/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs b/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs
index d0da1d5b..1283ecaf 100644
--- a/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs
+++ b/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs
@@ -15,10 +15,6 @@ namespace TensorFlowNET.UnitTest.Training
         [TestMethod]
         public void LinearRegression()
         {
-            var graph = tf.Graph().as_default();
-            var sess = new Session(graph);
-            sess.as_default();
-
             // Initialize the weights to `5.0` and the bias to `0.0`
             // In practice, these should be initialized to random values (for example, with `tf.random.normal`)
             var W = tf.Variable(5.0f);
diff --git a/test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs b/test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs
index 1632f1e7..d766890b 100644
--- a/test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs
+++ b/test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs
@@ -1,10 +1,10 @@
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using System;
+using Tensorflow;
 using Tensorflow.NumPy;
-using TensorFlowNET.UnitTest;
 using static Tensorflow.Binding;
 
-namespace Tensorflow.Keras.UnitTest.Optimizers
+namespace TensorFlowNET.UnitTest.Training
 {
     [TestClass]
     public class GradientDescentOptimizerTest : PythonTest