diff --git a/src/pystencils/kernelcreation.py b/src/pystencils/kernelcreation.py
index 154cb2307bdaaec0167632d4c77531632b1fe9e8..e31ef2fd42ec80b8c3c123cb20b11fa2a61c47df 100644
--- a/src/pystencils/kernelcreation.py
+++ b/src/pystencils/kernelcreation.py
@@ -32,7 +32,8 @@ from .backend.transformations import (
     SelectFunctions,
 )
 
-from .sympyextensions import AssignmentCollection, Assignment
+from .simp import AssignmentCollection
+from .assignment import Assignment
 
 
 __all__ = ["create_kernel"]