diff --git a/tests/parallel_test/cases.task b/tests/parallel_test/cases.task
index 6e0b180ed892531f4eb07c94b80a36c528473cd0..00e7c6ddd1b1617cc809f4974b6a847e3c97458f 100644
--- a/tests/parallel_test/cases.task
+++ b/tests/parallel_test/cases.task
@@ -120,6 +120,7 @@
 ,,y,system-test,./pytest.sh python3 ./test.py -f 0-others/fsync.py
 ,,n,system-test,python3 ./test.py -f 0-others/compatibility.py
 ,,n,system-test,python3 ./test.py -f 0-others/tag_index_basic.py
+,,n,system-test,python3 ./test.py -f 0-others/udfpy_main.py
 ,,y,system-test,./pytest.sh python3 ./test.py -f 1-insert/alter_database.py
 ,,y,system-test,./pytest.sh python3 ./test.py -f 1-insert/influxdb_line_taosc_insert.py
 ,,y,system-test,./pytest.sh python3 ./test.py -f 1-insert/opentsdb_telnet_line_taosc_insert.py
diff --git a/tests/system-test/0-others/udfpy/af_count.py b/tests/system-test/0-others/udfpy/af_count.py
index 285ef96b55ed83790181a3a7d871865bf116f287..226e02235ffebae116f245af79a57cc977853641 100644
--- a/tests/system-test/0-others/udfpy/af_count.py
+++ b/tests/system-test/0-others/udfpy/af_count.py
@@ -1,5 +1,3 @@
-import pickle
-
 def init():
     pass
 
@@ -7,23 +5,17 @@ def destroy():
     pass
 
 def start():
-    return pickle.dumps([])
+    return pickle.dumps(0)
 
 def finish(buf):
-    counts = pickle.loads(buf)
-    all_count = 0
-    for count in counts:
-        all_count += count
-
-    return all_count
+    count = pickle.loads(buf)
+    return count
 
 def reduce(datablock, buf):
     (rows, cols) = datablock.shape()
-    counts = pickle.loads(buf)
-    batch_count = 0
+    count = pickle.loads(buf)
     for i in range(rows):
         val = datablock.data(i, 0)
         if val is not None:
-           batch_count += 1
-    counts.append(batch_count)
-    return pickle.dumps(counts) 
+           count += 1
+    return pickle.dumps(count)
\ No newline at end of file
diff --git a/tests/system-test/0-others/udfpy_main.py b/tests/system-test/0-others/udfpy_main.py
index 55a60d4314819973ddf7f47655f288ac42356155..eaadfbdbd68cadef3c35a883f702cffa0c612228 100644
--- a/tests/system-test/0-others/udfpy_main.py
+++ b/tests/system-test/0-others/udfpy_main.py
@@ -401,8 +401,8 @@ class TDTestCase:
         # var
         stable = "meters"
         tbname = "d"
-        count = 3
-        rows =  3000000
+        count = 10
+        rows =  50000
         # do 
         self.create_table(stable, tbname, count)
         self.insert_data(tbname, rows)