diff --git a/{{cookiecutter.project_slug}}/tests/test_training.py b/{{cookiecutter.project_slug}}/tests/test_training.py
index cd9d1fb..22fa0e4 100644
--- a/{{cookiecutter.project_slug}}/tests/test_training.py
+++ b/{{cookiecutter.project_slug}}/tests/test_training.py
@@ -1,6 +1,6 @@
 import torch
 
-from project_template.training import NeuralNetwork
+from {{ cookiecutter.project_name }}.training import NeuralNetwork
 
 
 def test_neural_network():