diff --git a/etl/tests/test_csr_class.py b/etl/tests/test_csr_class.py
index db1524a..12ddf1c 100644
--- a/etl/tests/test_csr_class.py
+++ b/etl/tests/test_csr_class.py
@@ -4,7 +4,7 @@ from etl.surveyedData.surveryedData import surveyedDataProcessor
 from etl.db.db import get_db_session, init_db
 from etl.transform.types import AssessorInfo
 import pytest
-from etl.jjc_invoice import work_out_total_floor_area
+from etl.jjc_old_lewis_manual_way_ import work_out_total_floor_area
 
 @pytest.fixture(scope="module")
 def pre_site_note_path():
diff --git a/etl/tests/test_pre_site_note_class.py b/etl/tests/test_pre_site_note_class.py
index 58cf3e7..ec89a49 100644
--- a/etl/tests/test_pre_site_note_class.py
+++ b/etl/tests/test_pre_site_note_class.py
@@ -4,7 +4,7 @@ from etl.surveyedData.surveryedData import surveyedDataProcessor
 from etl.db.db import get_db_session, init_db
 from etl.transform.types import AssessorInfo
 import pytest
-from etl.jjc_invoice import work_out_total_floor_area
+from etl.jjc_old_lewis_manual_way_ import work_out_total_floor_area
 
 @pytest.fixture(scope="module")
 def pre_site_note_path():