diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 0142e00..a7f9250 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -13,7 +13,7 @@ jobs:
         with:
           activate-environment: dynamicscontrol
           environment-file: environment.yml
-          #python-version: 3.7
+          python-version: 3.11
           auto-activate-base: false
       - run: pip install .
       - run: |
diff --git a/environment.yml b/environment.yml
index 4880454..de731ba 100644
--- a/environment.yml
+++ b/environment.yml
@@ -12,6 +12,7 @@ dependencies:
   - slycot
   - pandas
   - keras
+  - tensorflow-base
   - scikit-learn
   - pytest
   - jupyter