diff --git a/template/tests/[% if compiled == true %]test_compiled.py[% endif %].jinja b/template/tests/[% if compiled == true %]test_compiled.py[% endif %].jinja
index 74cee46..b84bd36 100644
--- a/template/tests/[% if compiled == true %]test_compiled.py[% endif %].jinja	
+++ b/template/tests/[% if compiled == true %]test_compiled.py[% endif %].jinja	
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from ss_pybind11 import _core
+from {{ module_name }} import _core
 
 
 def test_core_version():