diff --git a/{{cookiecutter.directory_name}}/src/__init__.py b/{{cookiecutter.directory_name}}/src/__init__.py
new file mode 100644
index 0000000..8b13789
--- /dev/null
+++ b/{{cookiecutter.directory_name}}/src/__init__.py
@@ -0,0 +1 @@
+
diff --git a/{{cookiecutter.directory_name}}/src/main.py b/{{cookiecutter.directory_name}}/src/main.py
index 83d2334..f67bd8b 100644
--- a/{{cookiecutter.directory_name}}/src/main.py
+++ b/{{cookiecutter.directory_name}}/src/main.py
@@ -1,4 +1,4 @@
-from logging_utils import setup_logger
+from .logging_utils import setup_logger
 
 log = setup_logger()
 
diff --git a/{{cookiecutter.directory_name}}/tests/test_main.py b/{{cookiecutter.directory_name}}/tests/test_main.py
index 77155fa..d3f8b25 100644
--- a/{{cookiecutter.directory_name}}/tests/test_main.py
+++ b/{{cookiecutter.directory_name}}/tests/test_main.py
@@ -1,4 +1,13 @@
 # tests/test_main.py
+from pathlib import Path
+import sys
+
+# Ensure src is importable whether tests are run from the project root or template root
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
+
 def test_main_runs():
     from src.main import main
     assert main() is None  # or your actual expected behavior