diff --git a/test/test_array.py b/test/test_array.py
index 57ded681ce86f16554a7ca826b4d68b26b51290a..1f797539ad9ab52b5f4e5d8a6598b5e5dc7ec063 100644
--- a/test/test_array.py
+++ b/test/test_array.py
@@ -1171,7 +1171,7 @@ def test_nan_arithmetic(ctx_factory):
 def test_mem_pool_with_arrays(ctx_factory):
     context = ctx_factory()
     queue = cl.CommandQueue(context)
-    mem_pool = cl_tools.MemoryPool(cl_tools.CLAllocator(context))
+    mem_pool = cl_tools.MemoryPool(cl_tools.ImmediateAllocator(queue))
 
     a_dev = cl_array.arange(queue, 2000, dtype=np.float32, allocator=mem_pool)
     b_dev = cl_array.to_device(queue, np.arange(2000), allocator=mem_pool) + 4000