diff --git a/boundaries/boundaryhandling.py b/boundaries/boundaryhandling.py
index fb9225dfaf79f53fa756044041b4447b8da8b014..1f7b61f9f591efdbed38b682d8386cfaa0cafae7 100644
--- a/boundaries/boundaryhandling.py
+++ b/boundaries/boundaryhandling.py
@@ -1,7 +1,7 @@
 import numpy as np
 import sympy as sp
 from pystencils import TypedSymbol, create_indexed_kernel, Assignment
-from pystencils.backends.cbackend import CustomCppCode
+from pystencils.backends.cbackend import CustomCodeNode
 from pystencils.boundaries import BoundaryHandling
 from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo
 from pystencils.stencils import inverse_direction
@@ -76,7 +76,7 @@ class LatticeBoltzmannBoundaryHandling(BoundaryHandling):
                                                         boundary_obj, target=self._target, openmp=self._openmp)
 
 
-class LbmWeightInfo(CustomCppCode):
+class LbmWeightInfo(CustomCodeNode):
 
     # --------------------------- Functions to be used by boundaries --------------------------