diff --git a/tests/kernelcreation/test_gpu.py b/tests/kernelcreation/test_gpu.py
index 45acaf5f877772d3ccdb07c51ef2e4f70f236efe..f1905b1fcb7c7406f43cfb94af2928b6f35bc3f8 100644
--- a/tests/kernelcreation/test_gpu.py
+++ b/tests/kernelcreation/test_gpu.py
@@ -99,17 +99,17 @@ def test_indexing_options_2d(
     ast = create_kernel(asm, cfg)
     kernel = ast.compile()
 
-    src_arr = cp.ones((18, 34))
+    src_arr = cp.ones((18, 42))
     dst_arr = cp.zeros_like(src_arr)
 
     if manual_grid:
         match indexing_scheme:
             case "linear3d":
                 kernel.launch_config.block_size = (10, 8, 1)
-                kernel.launch_config.grid_size = (4, 4, 1)
+                kernel.launch_config.grid_size = (4, 2, 1)
             case "blockwise4d":
                 kernel.launch_config.block_size = (40, 1, 1)
-                kernel.launch_config.grid_size = (32, 1, 1)
+                kernel.launch_config.grid_size = (16, 1, 1)
 
     elif indexing_scheme == "linear3d":
         kernel.launch_config.block_size = (10, 8, 1)