diff --git a/test/TensorFlowNET.Keras.UnitTest/Gradient.cs b/test/TensorFlowNET.Keras.UnitTest/Gradient.cs
index fad8e118..f20eae0e 100644
--- a/test/TensorFlowNET.Keras.UnitTest/Gradient.cs
+++ b/test/TensorFlowNET.Keras.UnitTest/Gradient.cs
@@ -9,7 +9,7 @@ using Tensorflow.NumPy;
 namespace TensorFlowNET.Keras.UnitTest;
 
 [TestClass]
-public class GradientTest
+public class GradientTest : EagerModeTestBase
 {
     public IModel get_actor(int num_states)
     {