diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py
index 65be23065ea0bacf721af684ce7e67dc6a789f48..4d75f1ca68dce2e09b97fb1d6e226d7c351e34da 100644
--- a/src/pystencils/backend/kernelcreation/freeze.py
+++ b/src/pystencils/backend/kernelcreation/freeze.py
@@ -1,4 +1,3 @@
-from sympyextensions.reduction import ReducedAssignment
 from typing import overload, cast, Any
 from functools import reduce
 from operator import add, mul, sub, truediv
@@ -16,6 +15,7 @@ from ...sympyextensions import (
 )
 from ...sympyextensions.typed_sympy import TypedSymbol, CastFunc, DynamicType
 from ...sympyextensions.pointers import AddressOf, mem_acc
+from ...sympyextensions.reduction import ReducedAssignment
 from ...field import Field, FieldType
 
 from .context import KernelCreationContext
diff --git a/src/pystencils/simp/assignment_collection.py b/src/pystencils/simp/assignment_collection.py
index 4de3e8dc69663721dac224f3e39f52f1ebb78c47..212dbf75147db4d1889419e671091387badf545a 100644
--- a/src/pystencils/simp/assignment_collection.py
+++ b/src/pystencils/simp/assignment_collection.py
@@ -1,8 +1,6 @@
 import itertools
 from copy import copy
 
-from sympyextensions import reduced_assign
-from sympyextensions.reduction import ReducedAssignment
 from typing import Any, Dict, Iterable, Iterator, List, Optional, Sequence, Set, Union
 
 import sympy as sp
@@ -11,6 +9,8 @@ import pystencils
 from ..assignment import Assignment
 from .simplifications import (sort_assignments_topologically, transform_lhs_and_rhs, transform_rhs)
 from ..sympyextensions import count_operations, fast_subs
+from ..sympyextensions import reduced_assign
+from ..sympyextensions.reduction import ReducedAssignment
 
 
 class AssignmentCollection: