diff --git a/backend/app/services/lolbas_import_service.py b/backend/app/services/lolbas_import_service.py
index d4d80b4..1d675d5 100644
--- a/backend/app/services/lolbas_import_service.py
+++ b/backend/app/services/lolbas_import_service.py
@@ -60,6 +60,7 @@ from sqlalchemy.orm import Session
 # Import DataSource from app.models.data_source
 from app.models.data_source import DataSource
 from app.models.technique import Technique
+from app.models.test_template import TestTemplate
 from app.services.audit_service import log_action
 
 # Assign logger = logging.getLogger(__name__)
diff --git a/backend/app/services/sigma_import_service.py b/backend/app/services/sigma_import_service.py
index 96f213d..c19bbe6 100644
--- a/backend/app/services/sigma_import_service.py
+++ b/backend/app/services/sigma_import_service.py
@@ -57,6 +57,7 @@ from sqlalchemy.orm import Session
 
 # Import DataSource from app.models.data_source
 from app.models.data_source import DataSource
+from app.models.detection_rule import DetectionRule
 from app.models.technique import Technique
 from app.services.audit_service import log_action