diff --git a/tests/integration_tests/test_snowflake_cache.py b/tests/integration_tests/test_snowflake_cache.py
index 9781a49f..e92b89b3 100644
--- a/tests/integration_tests/test_snowflake_cache.py
+++ b/tests/integration_tests/test_snowflake_cache.py
@@ -9,13 +9,9 @@
 from collections.abc import Generator
 import os
 import sys
-import shutil
-from pathlib import Path
 
 import pytest
 
-from airbyte_cdk.models import ConfiguredAirbyteCatalog
-
 import airbyte as ab
 from airbyte import caches
 
diff --git a/tests/integration_tests/test_source_faker_integration.py b/tests/integration_tests/test_source_faker_integration.py
index c62c7d1a..ab9f5239 100644
--- a/tests/integration_tests/test_source_faker_integration.py
+++ b/tests/integration_tests/test_source_faker_integration.py
@@ -11,11 +11,11 @@
 import sys
 import shutil
 from pathlib import Path
+import typing
 
 import pytest
 import ulid
 
-from airbyte_cdk.models import ConfiguredAirbyteCatalog
 
 import airbyte as ab
 from airbyte import caches
@@ -130,7 +130,7 @@ def test_faker_pks(
 ) -> None:
     """Test that the append strategy works as expected."""
 
-    catalog: ConfiguredAirbyteCatalog = source_faker_seed_a.configured_catalog
+    catalog = source_faker_seed_a.configured_catalog
 
     assert catalog.streams[0].primary_key
     assert catalog.streams[1].primary_key