diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 41e15c5..b2d9cf2 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -9,19 +9,16 @@ on:
 jobs:
   test:
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        python-version: ["3.10"]
-      fail-fast: false
-    name: Run tests in mamba environment ${{ matrix.python-version }}
+    name: Run tests in mamba environment
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - name: Download mamba environment.yml
         run:
-          wget https://raw.githubusercontent.com/eWaterCycle/ewatercycle/main/environment.yml
+          wget https://raw.githubusercontent.com/eWaterCycle/ewatercycle/main/conda-lock.yml
       - uses: mamba-org/setup-micromamba@v1
         with:
-          environment-file: environment.yml
+          environment-file: conda-lock.yml
+          environment-name: ci
           cache-environment: true
           init-shell: bash
       - name: Install dependencies
diff --git a/pyproject.toml b/pyproject.toml
index 0d41572..7b1d95e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,7 +25,7 @@ classifiers = [
 
 # Include here only the dependencies for the eWaterCycle wrapped model
 dependencies = [
-  "ewatercycle>=2.0.0b2",
+  "ewatercycle>=2.3.1",
 ]
 dynamic = ["version"]
 
diff --git a/tests/test_forcing.py b/tests/test_forcing.py
index 4026958..14e1c65 100644
--- a/tests/test_forcing.py
+++ b/tests/test_forcing.py
@@ -134,6 +134,8 @@ def test_saved_yaml_content(self, forcing, tmp_path):
             """\
         start_time: '1989-01-02T00:00:00Z'
         end_time: '1999-01-02T00:00:00Z'
+        shape: Rhine.shp
+        filenames: {}
         forcing_file: marrmot.mat
         """
         )
@@ -143,7 +145,7 @@ def test_saved_yaml_content(self, forcing, tmp_path):
     def test_saved_yaml(self, forcing, tmp_path):
         saved_forcing = MarrmotForcing.load(tmp_path)
         # shape should is not included in the yaml file
-        forcing.shape = None
+        forcing.shape = forcing.directory / "Rhine.shp"
 
         assert forcing == saved_forcing
 
diff --git a/tests/test_model_m01.py b/tests/test_model_m01.py
index f2149bc..a0948b0 100644
--- a/tests/test_model_m01.py
+++ b/tests/test_model_m01.py
@@ -57,7 +57,9 @@ def test_str(self, model, forcing_file):
             [
                 "MarrmotForcing(start_time='1989-01-01T00:00:00Z', ",
                 f"end_time='1992-12-31T00:00:00Z', directory={repr(Path(forcing_file).parent)}, ",
-                "shape=None, forcing_file='BMI_testcase_m01_BuffaloRiver_TN_USA.mat')",
+                "shape=None, ",
+                "filenames={}, ",
+                "forcing_file='BMI_testcase_m01_BuffaloRiver_TN_USA.mat')",
             ]
         )
         expected = f"parameter_set=None forcing={expected_forcing}"