diff --git a/tests/pytest/query/query.py b/tests/pytest/query/query.py
index b076fe05eb9cac42fbca1717502402f39dba1961..7bec22b925ba41fd28a01dda84705d7263305382 100644
--- a/tests/pytest/query/query.py
+++ b/tests/pytest/query/query.py
@@ -13,10 +13,10 @@
 
 import sys
 import taos
-from util.log import tdLog
-from util.cases import tdCases
-from util.sql import tdSql
-from util.dnodes import tdDnodes
+from util.log import *
+from util.cases import *
+from util.sql import *
+from util.dnodes import *
 
 
 class TDTestCase: