diff --git a/tests/test_grid.py b/tests/test_grid.py
index 19a5e59..8f8d408 100644
--- a/tests/test_grid.py
+++ b/tests/test_grid.py
@@ -1,5 +1,5 @@
 import unittest
-from merra.grid import MERRACellgrid
+from merra.grid import create_merra_cell_grid
 
 class Test(unittest.TestCase):
     """
@@ -10,7 +10,7 @@ def test_MERRACellgrid(self):
         Test if the grid remains homogenous.
         :return:
         """
-        grid = MERRACellgrid()
+        grid = create_merra_cell_grid()
         assert grid.activegpis.size == 207936
         assert grid.activegpis[159290] == 159290
         assert grid.activearrcell[159290] == 1431