diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index beb3bb0..1184c24 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -99,6 +99,9 @@ jobs:
         if: startsWith(matrix.backend, 'none') != true
         run: |
           cd docs && make html
+      - name: Remove source directory to ensure that package from installation directory is used
+        run: |
+          rm -rf rbnicsx
       - name: Determine coverage and pytest options for unit tests
         id: unit_options
         run: |
diff --git a/pyproject.toml b/pyproject.toml
index 48cd25a..34e87aa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -109,6 +109,7 @@ line_length = 120
 multi_line_output = 4
 order_by_type = false
 combine_as_imports = true
+known_first_party = ["rbnicsx"]
 
 [tool.mypy]
 check_untyped_defs = true