diff --git a/tests/army/tools/benchmark/basic/bugs.py b/tests/army/tools/benchmark/basic/bugs.py
index aaf3312715..d1b45d95ed 100644
--- a/tests/army/tools/benchmark/basic/bugs.py
+++ b/tests/army/tools/benchmark/basic/bugs.py
@@ -30,7 +30,7 @@ def removeQuotation(origin):
 
     return value
 
-class TDTestCase:
+class TDTestCase(TBase):
     def caseDescription(self):
         """
         [TD-11510] taosBenchmark test cases
diff --git a/tests/army/tools/benchmark/basic/insertMix.py b/tests/army/tools/benchmark/basic/insertMix.py
index f9c08edf06..fc8b37f440 100644
--- a/tests/army/tools/benchmark/basic/insertMix.py
+++ b/tests/army/tools/benchmark/basic/insertMix.py
@@ -23,7 +23,7 @@ from frame.caseBase import *
 from frame import *
 
 
-class TDTestCase:
+class TDTestCase(TBase):
     def caseDescription(self):
         """
         [TD-13823] taosBenchmark test cases
diff --git a/tests/army/tools/benchmark/basic/queryMain.py b/tests/army/tools/benchmark/basic/queryMain.py
index 87c70c8df2..1d52163e31 100644
--- a/tests/army/tools/benchmark/basic/queryMain.py
+++ b/tests/army/tools/benchmark/basic/queryMain.py
@@ -37,7 +37,7 @@ def removeQuotation(origin):
 
     return value
 
-class TDTestCase:
+class TDTestCase(TBase):
     def caseDescription(self):
         """
         [TD-11510] taosBenchmark test cases
diff --git a/tests/army/tools/benchmark/basic/query_json.py b/tests/army/tools/benchmark/basic/query_json.py
index 62741a3401..6f71a508af 100644
--- a/tests/army/tools/benchmark/basic/query_json.py
+++ b/tests/army/tools/benchmark/basic/query_json.py
@@ -23,7 +23,7 @@ from frame.caseBase import *
 from frame import *
 
 
-class TDTestCase:
+class TDTestCase(TBase):
     def caseDescription(self):
         """
         [TD-11510] taosBenchmark test cases
diff --git a/tests/army/tools/benchmark/basic/stt.py b/tests/army/tools/benchmark/basic/stt.py
index b9288fcfa6..16affa9b27 100644
--- a/tests/army/tools/benchmark/basic/stt.py
+++ b/tests/army/tools/benchmark/basic/stt.py
@@ -23,7 +23,7 @@ from frame.caseBase import *
 from frame import *
 
 
-class TDTestCase:
+class TDTestCase(TBase):
     def caseDescription(self):
         """
         [TD-13823] taosBenchmark test cases