diff --git a/ingestion/src/metadata/ingestion/source/database/azuresql/query_parser.py b/ingestion/src/metadata/ingestion/source/database/azuresql/query_parser.py
index c3f860feaa3..a0411c6ad27 100644
--- a/ingestion/src/metadata/ingestion/source/database/azuresql/query_parser.py
+++ b/ingestion/src/metadata/ingestion/source/database/azuresql/query_parser.py
@@ -22,7 +22,7 @@ from metadata.generated.schema.entity.services.connections.metadata.openMetadata
 from metadata.generated.schema.metadataIngestion.workflow import (
     Source as WorkflowSource,
 )
-from metadata.ingestion.api.source import InvalidSourceException
+from metadata.ingestion.api.steps import InvalidSourceException
 from metadata.ingestion.source.database.query_parser_source import QueryParserSource
 
 
diff --git a/ingestion/tests/cli_e2e/base/test_cli_dashboard.py b/ingestion/tests/cli_e2e/base/test_cli_dashboard.py
index d8794c05dda..e4066834af2 100644
--- a/ingestion/tests/cli_e2e/base/test_cli_dashboard.py
+++ b/ingestion/tests/cli_e2e/base/test_cli_dashboard.py
@@ -19,7 +19,7 @@ from unittest import TestCase
 import pytest
 
 from metadata.ingestion.api.sink import SinkStatus
-from metadata.ingestion.api.source import SourceStatus
+from metadata.ingestion.api.status import Status
 
 from .e2e_types import E2EType
 from .test_cli import CliBase
@@ -95,21 +95,17 @@ class CliDashboardBase(TestCase):
             raise NotImplementedError()
 
         @abstractmethod
-        def assert_not_including(
-            self, source_status: SourceStatus, sink_status: SinkStatus
-        ):
+        def assert_not_including(self, source_status: Status, sink_status: SinkStatus):
             raise NotImplementedError()
 
         @abstractmethod
         def assert_for_vanilla_ingestion(
-            self, source_status: SourceStatus, sink_status: SinkStatus
+            self, source_status: Status, sink_status: SinkStatus
         ) -> None:
             raise NotImplementedError()
 
         @abstractmethod
-        def assert_filtered_mix(
-            self, source_status: SourceStatus, sink_status: SinkStatus
-        ):
+        def assert_filtered_mix(self, source_status: Status, sink_status: SinkStatus):
             raise NotImplementedError()
 
         @staticmethod
diff --git a/ingestion/tests/cli_e2e/base/test_cli_dbt.py b/ingestion/tests/cli_e2e/base/test_cli_dbt.py
index 70395431f61..85f1037db7b 100644
--- a/ingestion/tests/cli_e2e/base/test_cli_dbt.py
+++ b/ingestion/tests/cli_e2e/base/test_cli_dbt.py
@@ -21,7 +21,7 @@ import pytest
 from metadata.generated.schema.entity.data.table import Table
 from metadata.generated.schema.tests.testDefinition import TestDefinition, TestPlatform
 from metadata.ingestion.api.sink import SinkStatus
-from metadata.ingestion.api.source import SourceStatus
+from metadata.ingestion.api.status import Status
 
 from .test_cli import CliBase
 
@@ -107,12 +107,12 @@ class CliDBTBase(TestCase):
 
         @abstractmethod
         def assert_for_vanilla_ingestion(
-            self, source_status: SourceStatus, sink_status: SinkStatus
+            self, source_status: Status, sink_status: SinkStatus
         ) -> None:
             raise NotImplementedError()
 
         @abstractmethod
         def assert_for_dbt_ingestion(
-            self, source_status: SourceStatus, sink_status: SinkStatus
+            self, source_status: Status, sink_status: SinkStatus
         ) -> None:
             raise NotImplementedError()
diff --git a/ingestion/tests/cli_e2e/test_cli_snowflake.py b/ingestion/tests/cli_e2e/test_cli_snowflake.py
index bdc9528e77c..01d85337053 100644
--- a/ingestion/tests/cli_e2e/test_cli_snowflake.py
+++ b/ingestion/tests/cli_e2e/test_cli_snowflake.py
@@ -17,7 +17,7 @@ from typing import List
 import pytest
 
 from metadata.ingestion.api.sink import SinkStatus
-from metadata.ingestion.api.source import SourceStatus
+from metadata.ingestion.api.status import Status
 
 from .base.e2e_types import E2EType
 from .common.test_cli_db import CliCommonDB
@@ -79,7 +79,7 @@ class SnowflakeCliTest(CliCommonDB.TestSuite, SQACommonMethods):
         return "snowflake"
 
     def assert_for_vanilla_ingestion(
-        self, source_status: SourceStatus, sink_status: SinkStatus
+        self, source_status: Status, sink_status: SinkStatus
     ) -> None:
         self.assertTrue(len(source_status.failures) == 0)
         self.assertTrue(len(source_status.warnings) == 0)