diff --git a/tests/test_predictive_variable.py b/tests/test_predictive_variable.py
index 83a238ef..2a3ef478 100644
--- a/tests/test_predictive_variable.py
+++ b/tests/test_predictive_variable.py
@@ -4,7 +4,7 @@
 from helper_functions import load_and_format
 from pandas.testing import assert_frame_equal
 
-from src.predictive_variable import shift_by_strata_period
+from mbs_results.predictive_variable import shift_by_strata_period
 
 
 @pytest.fixture(scope="class")