diff --git a/test/test_array.py b/test/test_array.py
index d4ea8cd7c07d650492fd71b44bec9988658b1f5d..a65699f0d7b5da4c29cafc91b399dbbd1dae76ce 100644
--- a/test/test_array.py
+++ b/test/test_array.py
@@ -1168,7 +1168,7 @@ def test_multi_put(ctx_factory):
 
     cl_array.multi_put(cl_arrays, idx, out=out_arrays)
 
-    assert np.all(np.all(out_compare[i] == cl_arrays[i].get()) for i in range(10))
+    assert np.all(np.all(out_compare[i] == out_arrays[i].get()) for i in range(9))
 
 
 if __name__ == "__main__":