diff --git a/tests/test_migration_complete.py b/tests/test_migration_complete.py
index e9af8490..6a3a99c2 100644
--- a/tests/test_migration_complete.py
+++ b/tests/test_migration_complete.py
@@ -13,12 +13,10 @@ Tests are implemented for both PostgreSQL and Qdrant backends.
 """
 
 import pytest
-import asyncio
 import numpy as np
 from unittest.mock import MagicMock, patch, AsyncMock
 from lightrag.utils import EmbeddingFunc
 from lightrag.kg.qdrant_impl import QdrantVectorDBStorage, _find_legacy_collection
-from lightrag.kg.postgres_impl import PGVectorStorage
 
 
 # ============================================================================