diff --git a/test.py b/test.py
index 6317e3055d29701c5b7ddb62726d14db6282479e..da4bf4f9ae5668c45bc3402caa7d810a8bfedc98 100644
--- a/test.py
+++ b/test.py
@@ -62,8 +62,8 @@ def test_hoisted_grid_allocation(nxdirty, nydirty, nrow, nchan, epsilon):
     vis = baselines.ms2vis(ms, idx)
     user_grid = np.zeros((gconf.Nu(), gconf.Nv()),
                          dtype=np.complex128)
-    grid = ng.vis2grid_c(baselines, gconf, idx, vis, user_grid=None)
-    grid2 = ng.vis2grid_c(baselines, gconf, idx, vis, user_grid=user_grid)
+    grid = ng.vis2grid_c(baselines, gconf, idx, vis, grid_in=None)
+    grid2 = ng.vis2grid_c(baselines, gconf, idx, vis, grid_in=user_grid)
 
     # Almost same visibilities
     assert_array_almost_equal(grid, grid2)