diff --git a/test/test_train.py b/test/test_train.py
index 41b8b8a..a421946 100644
--- a/test/test_train.py
+++ b/test/test_train.py
@@ -2,7 +2,7 @@
 import pytest
 from tensorflow.keras.models import Sequential
 
-from train import read_data, create_model, train_model
+from ..train import read_data, create_model, train_model
 
 
 @pytest.fixture