diff --git a/src/pystencils/__init__.py b/src/pystencils/__init__.py
index 4f8b2660777dbe7a1fa535cfbbc2e64458a0d692..6cb375b61e63904c0cd3c2e6e9d6a3be86be6b29 100644
--- a/src/pystencils/__init__.py
+++ b/src/pystencils/__init__.py
@@ -18,7 +18,7 @@ from .field import Field, FieldType, fields
 from .types import create_type, create_numeric_type
 from .cache import clear_cache
 from .kernel_decorator import kernel, kernel_config
-from .kernelcreation import create_kernel, create_staggered_kernel
+from .codegen.driver import create_kernel, create_staggered_kernel
 from .codegen import Kernel
 from .jit import no_jit
 from .backend.exceptions import KernelConstraintsError