diff --git a/csst_common/wrapper.py b/csst_common/wrapper.py
index 525f6b58ec4498699a56c0c93aec72917f0884a0..368a2b0de6f8eaf8cde89b1add206a2eee71b86b 100644
--- a/csst_common/wrapper.py
+++ b/csst_common/wrapper.py
@@ -3,7 +3,7 @@ import time
 import traceback
 from collections import namedtuple
 
-from astropy import time
+from astropy.time import Time
 
 from csst_common.data_manager import CsstMsDataManager
 from csst_common.file_recorder import FileRecorder
@@ -93,7 +93,7 @@ def module_wrapper(func):
             # write time stamp
             if tstamp is not None:
                 with open(tstamp, "a+") as f:
-                    f.write(f"{time.Time.now().isot}+08:00\n")
+                    f.write(f"{Time.now().isot}+08:00\n")
             return ModuleResult(func.__name__, t_cost, status, fr, output)
 
     return call_module