diff --git a/benchmark.py b/benchmark.py
index 444b689e43a3a794ece2162c0511aceadb9f66c5..b48c19cacb9d410c1174a140fbb5992286042c97 100644
--- a/benchmark.py
+++ b/benchmark.py
@@ -17,15 +17,13 @@ from pyopencl.tools import (  # noqa
 from utilities import *
 
 
-def setup_random_array(*shape):
-    return np.random.random_sample(shape).astype(np.float32).copy(order="F")
-
-
-def setup_random_array_on_device(queue, *shape):
-    return cl.array.to_device(queue, setup_random_array(*shape))
-
-
 def benchmark_compute_flux_derivatives_gpu(ctx_factory):
+    def random_array_on_device(queue, *shape):
+        return cl.array.to_device(queue, random_array(*shape))
+    
+    def random_array(*shape):
+        return np.random.random_sample(shape).astype(np.float32).copy(order="F")
+
     logging.basicConfig(level="INFO")
 
     prg = get_weno_program()
@@ -41,10 +39,10 @@ def benchmark_compute_flux_derivatives_gpu(ctx_factory):
     nz = n
 
     print("ARRAY GEN")
-    states = setup_random_array_on_device(queue, nvars, nx+6, ny+6, nz+6)
-    fluxes = setup_random_array_on_device(queue, nvars, ndim, nx+6, ny+6, nz+6)
-    metrics = setup_random_array_on_device(queue, ndim, ndim, nx+6, ny+6, nz+6)
-    metric_jacobians = setup_random_array_on_device(queue, nx+6, ny+6, nz+6)
+    states = random_array_on_device(queue, nvars, nx+6, ny+6, nz+6)
+    fluxes = random_array_on_device(queue, nvars, ndim, nx+6, ny+6, nz+6)
+    metrics = random_array_on_device(queue, ndim, ndim, nx+6, ny+6, nz+6)
+    metric_jacobians = random_array_on_device(queue, nx+6, ny+6, nz+6)
     print("END ARRAY GEN")
 
     flux_derivatives_dev = cl.array.empty(queue, (nvars, ndim, nx+6, ny+6,