diff --git a/backend/app/services/caldera_import_service.py b/backend/app/services/caldera_import_service.py
index 43dccda..abb90d4 100644
--- a/backend/app/services/caldera_import_service.py
+++ b/backend/app/services/caldera_import_service.py
@@ -54,6 +54,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__)