diff --git a/lbmpy/boundaries/boundaries_in_kernel.py b/lbmpy/boundaries/boundaries_in_kernel.py
index 9375c33cd683b52bb5c7ec9c227d77bd15972507..6e01987b7253e29c97e1679340f313497121d421 100644
--- a/lbmpy/boundaries/boundaries_in_kernel.py
+++ b/lbmpy/boundaries/boundaries_in_kernel.py
@@ -1,8 +1,8 @@
 import sympy as sp
 
-from lbmpy.boundaries.boundaryhandling import LbmWeightInfo
 from lbmpy.advanced_streaming.indexing import BetweenTimestepsIndexing
 from lbmpy.advanced_streaming.utility import Timestep, get_accessor
+from lbmpy.custom_code_nodes import LbmWeightInfo
 from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo
 from pystencils.assignment import Assignment
 from pystencils.astnodes import Block, Conditional, LoopOverCoordinate, SympyAssignment