diff --git a/.github/workflows/main-ci.yml b/.github/workflows/main-ci.yml
index f5657df..9763dc6 100644
--- a/.github/workflows/main-ci.yml
+++ b/.github/workflows/main-ci.yml
@@ -37,5 +37,7 @@ jobs:
         python -c "import imgtools"
     - name: Run pytest
       run: |
-        pytest tests
+        pytest tests/test_components.py
+        pytest tests/test_modalities.py
+        pytest tests/test_pipeline.py
 
diff --git a/imgtools/io/__init__.py b/imgtools/io/__init__.py
index f3f2307..644d413 100644
--- a/imgtools/io/__init__.py
+++ b/imgtools/io/__init__.py
@@ -1,4 +1,3 @@
 from .common import *
 from .loaders import *
 from .writers import *
-from .dataset import *
diff --git a/setup.py b/setup.py
index 2bf6b73..99b60ca 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@
     
 setup(
     name="med-imagetools",
-    version="1.2.0",
+    version="1.2.0.1",
     author="Sejin Kim, Michal Kazmierski, Kevin Qu, Vishwesh Ramanathan, Benjamin Haibe-Kains",
     author_email="benjamin.haibe.kains@utoronto.ca",
     description="Transparent and reproducible image processing pipelines in Python.",