diff --git a/tests/pytest/util/common.py b/tests/pytest/util/common.py
index a290a92876..6384195a1b 100644
--- a/tests/pytest/util/common.py
+++ b/tests/pytest/util/common.py
@@ -18,7 +18,7 @@ import time
 import socket
 import json
 import toml
-from .boundary import DataBoundary
+from util.boundary import DataBoundary
 import taos
 from util.log import *
 from util.sql import *
diff --git a/tests/system-test/1-insert/mutil_stage.py b/tests/system-test/1-insert/mutil_stage.py
index 6584c4081b..e4342eca24 100644
--- a/tests/system-test/1-insert/mutil_stage.py
+++ b/tests/system-test/1-insert/mutil_stage.py
@@ -210,7 +210,8 @@ class TDTestCase:
             tdSql.execute(f'create table {dbname}.{ctb_pre}{i+1} using {dbname}.{stb} tags ( {i+1} )')
 
     def __insert_data(self, rows,dbname=DBNAME):
-        data = DataSet().get_order_set(rows)
+        data = DataSet()
+        data.get_order_set(rows)
 
         tdLog.printNoPrefix("==========step: start inser data into tables now.....")
         for i in range(self.rows):