diff --git a/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs b/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs
index a37f2892..d0da1d5b 100644
--- a/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs
+++ b/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs
@@ -15,7 +15,9 @@ namespace TensorFlowNET.UnitTest.Training
         [TestMethod]
         public void LinearRegression()
         {
-            tf.Graph().as_default();
+            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`)