diff --git a/tests/fixtures/fixtures.py b/tests/fixtures/fixtures.py
index 393a6b9..f898df6 100644
--- a/tests/fixtures/fixtures.py
+++ b/tests/fixtures/fixtures.py
@@ -7,7 +7,7 @@ sys.path.append(".")
 import pytest
 from IPython.core.interactiveshell import InteractiveShell
 
-from v1.v1 import NVCCPlugin
+from nvcc4jupyter.plugin import NVCCPlugin
 
 
 @pytest.fixture(scope="session")
diff --git a/tests/test_v1.py b/tests/test_v1.py
index 255ffa6..45f5d4c 100644
--- a/tests/test_v1.py
+++ b/tests/test_v1.py
@@ -11,7 +11,7 @@ import pytest
 sys.path.append(".")
 
 
-from v1.v1 import NVCCPlugin
+from nvcc4jupyter.plugin import NVCCPlugin
 
 
 def check_profiler_output(output: str):