diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 818bf94..2f2a6b7 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -46,10 +46,8 @@ 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.1" or ${{ matrix.tensorflow-version }} == "2.17.0"; then
-            pip install tf-keras
-            export TF_USE_LEGACY_KERAS=1
-          fi
+          pip install tf-keras
+          export TF_USE_LEGACY_KERAS=1
           pip install coveralls
           pip install -r requirements.txt