diff --git a/lbmpy/sparse/mapping.py b/lbmpy/sparse/mapping.py
index 4c822f31df43ddcbcab98db126a7887016589dd8..664907c73168cc25ea88d4b4df55f9ce79341879 100644
--- a/lbmpy/sparse/mapping.py
+++ b/lbmpy/sparse/mapping.py
@@ -3,7 +3,7 @@ from typing import Tuple
 import numpy as np
 import sympy as sp
 
-from lbmpy.boundaries.boundaryhandling import LbmWeightInfo
+from lbmpy.advanced_streaming.custom_code_nodes import LbmWeightInfo
 from pystencils import Assignment, Field, TypedSymbol
 from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo
 from pystencils.boundaries.createindexlist import (