diff --git a/python/tests/rd_tests/test_grid_equinor.py b/python/tests/rd_tests/test_grid_equinor.py
index 033485ebd..5b7941064 100755
--- a/python/tests/rd_tests/test_grid_equinor.py
+++ b/python/tests/rd_tests/test_grid_equinor.py
@@ -12,7 +12,7 @@
 import time
 from resdata import ResDataType, UnitSystem
 from resdata.resfile import ResdataKW, ResdataFile, openResdataFile
-from resdata.grid import Grid
+from resdata.grid import Grid, GridGenerator
 from resdata.util.util import DoubleVector, IntVector
 from resdata.util.test import TestAreaContext
 from tests import ResdataTest, equinor_test