diff --git a/conftest.py b/conftest.py
index 23ed4d49bec9c4bf4f6eec5daa573be7bd930475..b25dbeca4f158fb7e5bb55a529174eb5d1b14f8d 100644
--- a/conftest.py
+++ b/conftest.py
@@ -16,7 +16,7 @@ from lbmpy._compat import IS_PYSTENCILS_2
 # Trigger config file reading / creation once - to avoid race conditions when multiple instances are creating it
 # at the same time
 if IS_PYSTENCILS_2:
-    from pystencils.backend.jit import legacy_cpu
+    from pystencils.jit import legacy_cpu
 else:
     from pystencils.cpu import cpujit
 
diff --git a/src/lbmpy/creationfunctions.py b/src/lbmpy/creationfunctions.py
index dcdbdd1c2cd58a8cd8e7b4c0a94fdb9b928c419d..1883a1e2b8d129b2573d0919fda3714f3e1d1856 100644
--- a/src/lbmpy/creationfunctions.py
+++ b/src/lbmpy/creationfunctions.py
@@ -93,7 +93,7 @@ from lbmpy.methods.abstractlbmethod import LbmCollisionRule, AbstractLbMethod
 from lbmpy.methods.cumulantbased import CumulantBasedLbMethod
 
 if IS_PYSTENCILS_2:
-    from pystencils import KernelFunction
+    from pystencils import Kernel as KernelFunction
 else:
     from pystencils.astnodes import KernelFunction