diff --git a/tests/system-test/1-insert/mutil_stage.py b/tests/system-test/1-insert/mutil_stage.py
index 842e5a5167075bf8c1151a9f261ca16a8a5d92e8..d5c1c5d690d2e003c2886e12afd10e97295384d2 100644
--- a/tests/system-test/1-insert/mutil_stage.py
+++ b/tests/system-test/1-insert/mutil_stage.py
@@ -1,4 +1,4 @@
-import datetime
+from datetime import datetime
 
 from typing import List, Any, Tuple
 from util.log import *
@@ -35,7 +35,7 @@ ALL_COL = [PRIMARY_COL, INT_COL, BINT_COL, SINT_COL, TINT_COL, FLOAT_COL, DOUBLE
 TAG_COL = [INT_TAG]
 # insert data args：
 TIME_STEP = 10000
-NOW = int(datetime.datetime.timestamp(datetime.datetime.now()) * 1000)
+NOW = int(datetime.timestamp(datetime.now()) * 1000)
 
 # init db/table
 DBNAME  = "db"