diff --git a/pystencils_benchmark/benchmark.py b/pystencils_benchmark/benchmark.py
index 67cc3d58a911371e9db8ce8374cf69c36b8bffdf..34bdf6f8704cf4366f9a1e03ef5e4130ae17a302 100644
--- a/pystencils_benchmark/benchmark.py
+++ b/pystencils_benchmark/benchmark.py
@@ -6,9 +6,9 @@ from jinja2 import Environment, PackageLoader, StrictUndefined
 import numpy as np
 
 from pystencils.backends.cbackend import generate_c, get_headers
-from pystencils.astnodes import KernelFunction, PragmaBlock
+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.integer_functions import modulo_ceil