diff --git a/ingestion/tests/integration/automations/test_connection_automation.py b/ingestion/tests/integration/automations/test_connection_automation.py
index dad414d99a22..f93083aa606c 100644
--- a/ingestion/tests/integration/automations/test_connection_automation.py
+++ b/ingestion/tests/integration/automations/test_connection_automation.py
@@ -27,7 +27,6 @@
     WorkflowStatus,
     WorkflowType,
 )
-from metadata.generated.schema.entity.data.pipeline import StatusType
 from metadata.generated.schema.entity.services.connections.database.common.basicAuth import (
     BasicAuth,
 )
@@ -38,6 +37,9 @@
 from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import (
     OpenMetadataConnection,
 )
+from metadata.generated.schema.entity.services.connections.testConnectionResult import (
+    StatusType,
+)
 from metadata.generated.schema.entity.services.databaseService import DatabaseConnection
 from metadata.generated.schema.entity.services.serviceType import ServiceType
 from metadata.generated.schema.security.client.openMetadataJWTClientConfig import (