diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 89b3e46..818bf94 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -17,6 +17,7 @@ jobs:
       fail-fast: true
 
       matrix:
+
         python-version: ["3.9", "3.10", "3.12"]
         tensorflow-version: ["2.15.0", "2.16.1", "2.17.0"]
         tfp-version: ["0.23.0", "0.24.0"]
@@ -45,7 +46,7 @@ jobs:
           python -m pip install --upgrade pip
           pip install tensorflow==${{ matrix.tensorflow-version }}
           pip install tensorflow-probability==${{ matrix.tfp-version }}
-          if ${{ matrix.tensorflow-version }} >= "2.16.0"; then
+          if ${{ matrix.tensorflow-version }} == "2.16.1" or ${{ matrix.tensorflow-version }} == "2.17.0"; then
             pip install tf-keras
             export TF_USE_LEGACY_KERAS=1
           fi
diff --git a/tensiometer/synthetic_probability/synthetic_probability.py b/tensiometer/synthetic_probability/synthetic_probability.py
index f3f8ebf..2a3a813 100644
--- a/tensiometer/synthetic_probability/synthetic_probability.py
+++ b/tensiometer/synthetic_probability/synthetic_probability.py
@@ -1328,7 +1328,6 @@ def save(self, outroot):
         # save out trainable transformation:
         if self.trainable_transformation is not None:
             self.trainable_transformation.save(outroot)
-        
         #
         return None