diff --git a/pytest/test_func_ge.py b/pytest/test_func_ge.py
index 8984610..e2bace1 100644
--- a/pytest/test_func_ge.py
+++ b/pytest/test_func_ge.py
@@ -1,5 +1,5 @@
 from GENetLib.sim_data_func import sim_data_func
-from GENetLib.sim_data_func import func_ge
+from GENetLib.func_ge import func_ge
 
 def test_func_ge():
   func_continuous = sim_data_func(n=50, m=30, ytype='Continuous', seed=123)