diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index ab0a709..7f7fb4c 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -33,6 +33,7 @@ jobs:
           pip install wheel
           pip install -r requirements.txt --prefer-binary
           pip install -e . --no-deps
+          pip install bsb-test==4.0.0rc3 --no-deps
       - name: Run tests & coverage
         run: |
           coverage run -p -m unittest
diff --git a/test/test_chunks.py b/test/test_chunks.py
index d6b30ee..d695a00 100644
--- a/test/test_chunks.py
+++ b/test/test_chunks.py
@@ -1,8 +1,7 @@
 import unittest
 
 import numpy as np
-from bsb.core import Scaffold
-from bsb.storage import Chunk
+from bsb import Chunk, Scaffold
 from bsb_test import NumpyTestCase, RandomStorageFixture, skip_parallel, timeout
 from bsb_test.configs import get_test_config
 
diff --git a/test/test_cs.py b/test/test_cs.py
index abd7c3a..0366b63 100644
--- a/test/test_cs.py
+++ b/test/test_cs.py
@@ -1,7 +1,6 @@
 import unittest
 
-from bsb.core import Scaffold
-from bsb.services.pool import WorkflowError
+from bsb import Scaffold, WorkflowError
 from bsb_test import FixedPosConfigFixture, NumpyTestCase, RandomStorageFixture
 
 from bsb_hdf5.connectivity_set import LocationOutOfBoundsError
diff --git a/test/test_mr.py b/test/test_mr.py
index 3a3b209..a718dd2 100644
--- a/test/test_mr.py
+++ b/test/test_mr.py
@@ -4,8 +4,7 @@
 
 import h5py
 import numpy as np
-from bsb.services import MPI
-from bsb.storage import Storage
+from bsb import MPI, Storage
 
 
 class TestHandcrafted(unittest.TestCase):