diff --git a/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py b/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py
index 75781881846c24..5afe21531a296a 100644
--- a/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py
+++ b/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py
@@ -1,6 +1,7 @@
-import pymysql
 import time
 
+import pymysql
+
 
 def create_table():
     try: