diff --git a/api/tests/test_containers_integration_tests/services/test_webhook_service.py b/api/tests/test_containers_integration_tests/services/test_webhook_service.py
index e1056718e6..36776778e7 100644
--- a/api/tests/test_containers_integration_tests/services/test_webhook_service.py
+++ b/api/tests/test_containers_integration_tests/services/test_webhook_service.py
@@ -8,7 +8,7 @@ from flask import Flask
 from werkzeug.datastructures import FileStorage
 
 from models.model import App
-from models.workflow import Workflow, WorkflowWebhookTrigger
+from models.trigger import WorkflowWebhookTrigger
 from services.account_service import AccountService, TenantService
 from services.trigger.webhook_service import WebhookService