diff --git a/csst_ifs_sim/csst_ifs_sim.py b/csst_ifs_sim/csst_ifs_sim.py
index 4c4349a40a7c21ceb008369895412e2f6773ce4a..b1628bcb295e3517bbde41ce49b41a7b224be023 100644
--- a/csst_ifs_sim/csst_ifs_sim.py
+++ b/csst_ifs_sim/csst_ifs_sim.py
@@ -2327,8 +2327,8 @@ class IFSsimulator():
         crImage_r = np.zeros((3072, 6144), dtype=float)
 
         # cosmic ray instance
-        cosmics_b = cosmicrays.cosmicrays(self.log, crImage_b, crInfo=self.cr)
-        cosmics_r = cosmicrays.cosmicrays(self.log, crImage_r, crInfo=self.cr)
+        cosmics_b = cosmicrays(self.log, crImage_b, crInfo=self.cr)
+        cosmics_r = cosmicrays(self.log, crImage_r, crInfo=self.cr)
 
         CCD_cr_b = cosmics_b.addUpToFraction(
             self.information['coveringfraction']*self.information['exptime']/300.0, limit=None)