diff --git a/.github/actions/install-env/action.yml b/.github/actions/install-env/action.yml
index 8c1c91cd31..2d76956886 100644
--- a/.github/actions/install-env/action.yml
+++ b/.github/actions/install-env/action.yml
@@ -33,6 +33,7 @@ runs:
         installer-parallel: true
 
     - name: Load cached virtual env
+      id: cached-poetry-dependencies
       uses: actions/cache@v3
       with:
         path: .venv
diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
index fa8e981ac5..82055ac169 100644
--- a/.github/workflows/unit-tests.yml
+++ b/.github/workflows/unit-tests.yml
@@ -14,39 +14,10 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - name: Set up Python
-        id: set-up-python
-        uses: actions/setup-python@v4
+      - name: Build River
+        uses: ./.github/actions/install-env
         with:
-          python-version: 3.11
-
-      - name: Load cached Poetry installation
-        uses: actions/cache@v3
-        with:
-          path: ~/.local # the path depends on the OS
-          key: poetry-1 # increment to reset cache
-
-      - name: Install poetry
-        uses: snok/install-poetry@v1
-        with:
-          virtualenvs-create: true
-          virtualenvs-in-project: true
-          installer-parallel: true
-
-      - name: Load cached virtual env
-        id: cached_poetry_install
-        uses: actions/cache@v3
-        with:
-          path: .venv
-          key: venv-${{ runner.os }}-${{ steps.set-up-python.outputs.python-version }}-${{ hashFiles('**/poetry.lock') }}
-
-      - name: Install dependencies
-        shell: bash
-        if: steps.cached_poetry_install.outputs.cache-hit != 'true'
-        run: poetry install --no-interaction --no-ansi
-
-      - name: See pyx
-        run: ls /home/runner/work/river/river/.venv/lib/python3.11/site-packages/river-0.19.0.dist-info/RECORD river/utils/
+          python-version: "3.12"
 
       - name: Cache River datasets
         uses: actions/cache@v3
@@ -60,15 +31,15 @@ jobs:
           path: ~/scikit_learn_data
           key: ${{ runner.os }}
 
-      #   - name: Download datasets
-      #     run: |
-      #       poetry run python -c "from river import datasets; datasets.CreditCard().download(); datasets.Elec2().download(); datasets.SMSSpam().download()"
-      #       poetry run python -c "from river import bandit; bandit.datasets.NewsArticles().download()"
+      - name: Download datasets
+        run: |
+          poetry run python -c "from river import datasets; datasets.CreditCard().download(); datasets.Elec2().download(); datasets.SMSSpam().download()"
+          poetry run python -c "from river import bandit; bandit.datasets.NewsArticles().download()"
 
       - name: pytest [Branch]
         if: github.event_name == 'pull_request'
         run: |
-          poetry run pytest river/linear_model/lin_reg.py --durations=10 -n logical # Run pytest on all logical CPU cores
+          poetry run pytest --durations=10 -n logical # Run pytest on all logical CPU cores
 
       - name: pytest [Main]
         if: github.event_name == 'push'
diff --git a/pyproject.toml b/pyproject.toml
index 0da1ab6859..d6e023a1b6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,8 +9,8 @@ description = "Online machine learning in Python"
 authors = ["Max Halford <maxhalford25@gmail.com>"]
 
 [tool.poetry.build]
-script = "build.py"
 generate-setup-file = true
+script = "build.py"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<3.13"