diff --git a/src/dags/libs/create_records.py b/src/dags/libs/create_records.py
index d219677b34f9038fd82cf677ea58af590bab7cda..19995af843fc2cfdb63f6615064c143fefd17eda 100644
--- a/src/dags/libs/create_records.py
+++ b/src/dags/libs/create_records.py
@@ -21,8 +21,8 @@ import logging
 from airflow.models import Variable
 from libs.refresh_token import AirflowTokenRefresher, refresh_token
 from osdu_api.model.acl import Acl
-from osdu_api.model.legal import Legal
-from osdu_api.model.legal_compliance import LegalCompliance
+from osdu_api.model.legal.legal import Legal
+from osdu_api.model.legal.legal_compliance import LegalCompliance
 from osdu_api.model.record import Record
 from osdu_api.model.record_ancestry import RecordAncestry
 from osdu_api.storage.record_client import RecordClient