diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 9742ae8..e28ed46 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -28,11 +28,11 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         python -m pip install poetry
-        poetry install .
+        poetry install
     - name: Lint with flake8
       run: |
         # stop the build if there are Python syntax errors or undefined names
-        flake8 . --count --show-source --statistics
+        poetry run flake8 src/pydase --count --show-source --statistics
     - name: Test with pytest
       run: |
-        pytest
+        poetry run pytest
diff --git a/src/pydase/server/__init__.py b/src/pydase/server/__init__.py
index 5ccd59f..09ed39c 100644
--- a/src/pydase/server/__init__.py
+++ b/src/pydase/server/__init__.py
@@ -1,4 +1,3 @@
 from .server import Server
-from .web_server import WebAPI
 
 __all__ = ["Server"]