diff --git a/justfile b/justfile
index e197f93..09e812f 100644
--- a/justfile
+++ b/justfile
@@ -1,5 +1,5 @@
 test +args="":
-    uv run pytest -vv --color=yes --showlocals --no-header '{{ args }}'
+    uv run pytest -vv --color=yes --showlocals '{{ args }}'
 
 format:
     uv run ruff format .
diff --git a/tests/conftest.py b/tests/conftest.py
index bad49dd..2a3e000 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -218,7 +218,6 @@ def generate_no_import(self) -> bool:
 
         if str(self.src_path) not in sys.path:
             sys.path.insert(0, str(self.src_path))
-        importlib.invalidate_caches()
 
         return generate_sql_package(
             schema_path=Path("schema.sql"),
@@ -232,9 +231,8 @@ def generate_no_import(self) -> bool:
     def generate(self) -> Any:
         self.generate_no_import()
 
-        # Force fresh import
-        if self.pkg_name in sys.modules:
-            sys.modules.pop(self.pkg_name)
+        importlib.invalidate_caches()
+        sys.modules.pop(self.pkg_name, None)
 
         mod = importlib.import_module(self.pkg_name)
         self.generated_modules.append(mod)