diff --git a/tests/test_cube3d.py b/tests/test_cube3d.py
index 2f9706af45ac513ee97e220bd8ec2f3163df8574..c60d6423de5fb27a53dcb7b6564f71f350f33bd3 100644
--- a/tests/test_cube3d.py
+++ b/tests/test_cube3d.py
@@ -1,10 +1,10 @@
 import sys
 sys.path.insert(0, '..')
 import unittest
-from gehong import spec1d as s
-from gehong import map2d as m
-from gehong import cube3d as b
-from gehong import config as c
+from csst_ifs_gehong import spec1d as s
+from csst_ifs_gehong import map2d as m
+from csst_ifs_gehong import cube3d as b
+from csst_ifs_gehong import config as c
 import numpy as np
 import matplotlib.pyplot as plt
 
@@ -76,4 +76,4 @@ class test_cube3d(unittest.TestCase):
         print(">>> Unit test for **Spec Cube** Modelling has been passed!")
 
 if __name__=='__main__':
-    unittest.main()
\ No newline at end of file
+    unittest.main()
diff --git a/tests/test_map2d.py b/tests/test_map2d.py
index 7ed6cbc34a841e71362e3f70a15c1a04c5d2d7a0..0305204c54db0163fd4eafd0749a39f914ecb48f 100644
--- a/tests/test_map2d.py
+++ b/tests/test_map2d.py
@@ -1,8 +1,8 @@
 import sys
 sys.path.insert(0, '..')
 import unittest
-from gehong import map2d as m
-from gehong import config as c
+from csst_ifs_gehong import map2d as m
+from csst_ifs_gehong import config as c
 import numpy as np
 import matplotlib.pyplot as plt
 
@@ -234,4 +234,4 @@ class test_map2d(unittest.TestCase):
             print("SBmap No Problem!")
 
 if __name__=='__main__':
-    unittest.main()
\ No newline at end of file
+    unittest.main()
diff --git a/tests/test_spec1d.py b/tests/test_spec1d.py
index 4daba171d7d9c86ce602ea220c13a52cb764e6cc..abe81173f7f2c24bf7e8b0bc8ebd33ef963124b8 100644
--- a/tests/test_spec1d.py
+++ b/tests/test_spec1d.py
@@ -1,8 +1,8 @@
 import sys
 sys.path.insert(0, '..')
 import unittest
-from gehong import spec1d as s
-from gehong import config as c
+from csst_ifs_gehong import spec1d as s
+from csst_ifs_gehong import config as c
 import numpy as np
 import matplotlib.pyplot as plt
 
@@ -365,4 +365,4 @@ class test_spec1d(unittest.TestCase):
         print(">>> Unit test for **AGN** Modelling has been passed!")
 
 if __name__=='__main__':
-    unittest.main()
\ No newline at end of file
+    unittest.main()