diff --git a/.github/workflows/python-pytest.yml b/.github/workflows/python-pytest.yml
index 67e2be3..38d1fca 100644
--- a/.github/workflows/python-pytest.yml
+++ b/.github/workflows/python-pytest.yml
@@ -18,15 +18,11 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: [3.9, 3.10]
-        tf-version: ["2.6.4", "2.7.4", "2.8.4", "2.9.3"]
-        include:
-          - python-version: "3.10"
-            tf-version: "2.10"
-          - python-version: "3.10"
-            tf-version: "2.11"
-          - python-version: "3.10"
-            tf-version: "2.17"
+        python-version: ["3.9", "3.10"]
+        tf-version: ["2.15", "2.16", "2.17"]
+#        include:
+#          - python-version: "3.10"
+#            tf-version: "2.17"
 
     steps:
     - uses: actions/checkout@v2
diff --git a/deeptables/tests/misc.py b/deeptables/tests/misc.py
index db4809f..0fc79db 100644
--- a/deeptables/tests/misc.py
+++ b/deeptables/tests/misc.py
@@ -6,7 +6,7 @@
 """
 
 """
-#
+# https://github.com/tensorflow/tensorflow/issues/39554
 def cast(var, dtype):
     if isinstance(var, tf.Tensor):
         return tf.cast(var, dtype=dtype)