diff --git a/pystencils_benchmark/benchmark_gpu.py b/pystencils_benchmark/benchmark_gpu.py
index d68a31dd685f0512bf5d9a7f8a7758939d65d732..1c4e24c4a543f4afdb28d4c96ef967c407d50d77 100644
--- a/pystencils_benchmark/benchmark_gpu.py
+++ b/pystencils_benchmark/benchmark_gpu.py
@@ -6,10 +6,10 @@ from jinja2 import Environment, PackageLoader, StrictUndefined
 from pystencils.backends.cbackend import generate_c, get_headers
 from pystencils.astnodes import KernelFunction
 from pystencils.enums import Backend
-from pystencils.data_types import get_base_type
+from pystencils.typing import get_base_type
 from pystencils.sympyextensions import prod
-from pystencils.transformations import get_common_shape
-from pystencils.gpucuda import BlockIndexing
+from pystencils.transformations import get_common_field
+# from pystencils.gpucuda import BlockIndexing
 
 from pystencils_benchmark.enums import Compiler
 
@@ -121,7 +121,7 @@ def kernel_main(kernels_ast: List[KernelFunction], timing: bool = True, cuda_blo
                 fields.append((p.field_name, dtype, elements))
                 call_parameters.append(p.field_name)
 
-            common_shape = get_common_shape(kernel.fields_accessed)
+            common_shape = get_common_field(kernel.fields_accessed).shape
             indexing = kernel.indexing
             block_and_thread_numbers = indexing.call_parameters(common_shape)
             block_and_thread_numbers['block'] = tuple(int(i) for i in block_and_thread_numbers['block'])