diff --git a/pytest/test_grid_func_ge.py b/pytest/test_grid_func_ge.py
index 562ff03..10e30ef 100644
--- a/pytest/test_grid_func_ge.py
+++ b/pytest/test_grid_func_ge.py
@@ -1,3 +1,4 @@
+import pytest
 from GENetLib.sim_data_func import sim_data_func
 from GENetLib.grid_func_ge import grid_func_ge