diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index db30c05..aa9aaee 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -14,7 +14,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: ["ubuntu", "windows"]
+        os: ["ubuntu"]
     defaults:
       run:
         shell: bash -l {0}
@@ -30,7 +30,7 @@ jobs:
         with:
           miniconda-version: "latest"
           environment-file: environment.yaml
-          activate-environment: test
+          activate-environment: reservoir
 
       # Install testing dependencies
       - name: Install testing dependencies
@@ -40,4 +40,4 @@ jobs:
       # Run tests
       - name: Run tests with pytest
         run: |
-          python -m pytest -v tests/
+          python -m pytest -v tests/test_predict.py
diff --git a/amn/__main__.py b/amn/__main__.py
index e496d81..6853b48 100644
--- a/amn/__main__.py
+++ b/amn/__main__.py
@@ -234,7 +234,7 @@ def pred_amn (argv=None):
                 if len(extracted_model) != 1:
                     raise FileNotFoundError(f"Expected exactly one .h5 file, found {len(extracted_model)}")
                 if len(extracted_cobra) != 1:
-                    raise FileNotFoundError(f"Expected exactly one .h5 file, found {len(extracted_cobra)}")
+                    raise FileNotFoundError(f"Expected exactly one .xlm file, found {len(extracted_cobra)}")
 
 
             # Call prediction using extracted files
diff --git a/environment.yaml b/environment.yaml
index 4b3690d..8a765d8 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -5,7 +5,7 @@ channels:
 
 dependencies:
   - python =3.8  
-  - numpy
+  - numpy =1.23.5
   - pandas
   - matplotlib
   - scikit-learn
@@ -16,6 +16,6 @@ dependencies:
   - typing_extensions =4.5.0
   - pip
   - ipykernel
-  - tensorflow =2.13.0
+  - tensorflow =2.4.1
   - tensorflow-probability
   - importlib-resources
\ No newline at end of file