diff --git a/tests/test_tfmad.py b/tests/test_tfmad.py
index b045288c9b0bd1926cbb9d6766197e8c3c4b81f3..c1a28f5bd53aa042804211b994f2aa8d3e3b9d7b 100644
--- a/tests/test_tfmad.py
+++ b/tests/test_tfmad.py
@@ -59,6 +59,7 @@ def test_tfmad_two_stencils():
 @pytest.mark.skipif("NO_TENSORFLOW_TEST" in os.environ, reason="Requires Tensorflow")
 def test_tfmad_gradient_check():
     a, b, out = ps.fields("a, b, out: double[21,13]")
+    print(a.shape)
 
     cont = ps.fd.Diff(a, 0) - ps.fd.Diff(a, 1) - ps.fd.Diff(b, 0) + ps.fd.Diff(b, 1)
     discretize = ps.fd.Discretization2ndOrder(dx=1)
@@ -248,12 +249,12 @@ def test_tfmad_two_outputs():
 
 
 def main():
-    test_tfmad_stencil()
-    test_tfmad_two_stencils()
-    test_tfmad_gradient_check_torch()
+    # test_tfmad_stencil()
+    # test_tfmad_two_stencils()
+    # test_tfmad_gradient_check_torch()
     test_tfmad_gradient_check()
-    test_tfmad_vector_input_data()
-    test_tfmad_two_outputs()
+    # test_tfmad_vector_input_data()
+    # test_tfmad_two_outputs()
 
 
 if __name__ == '__main__':