diff --git a/src/pystencils_autodiff/lbm/__init__.py b/src/pystencils_autodiff/lbm/__init__.py
index a28beebcc67532ac1db816dfc49497edd4a0e301..4c1cb27e87ecad44589be4a8b88e7d0914466d2b 100644
--- a/src/pystencils_autodiff/lbm/__init__.py
+++ b/src/pystencils_autodiff/lbm/__init__.py
@@ -1,4 +1,4 @@
-from lbmpy.autodiff._autodiff_lbstep import AutoDiffLatticeBoltzmannStep
-from lbmpy.autodiff.adjoint_boundaryconditions import AdjointBoundaryCondition
+from pystencils_autodiff.lbm.adjoint_boundaryconditions import AdjointBoundaryCondition
+from pystencils_autodiff.lbm._autodiff_lbstep import AutoDiffLatticeBoltzmannStep
 
 __all__ = ['AutoDiffLatticeBoltzmannStep', 'AdjointBoundaryCondition']
diff --git a/src/pystencils_autodiff/lbm/_autodiff_lbstep.py b/src/pystencils_autodiff/lbm/_autodiff_lbstep.py
index 0cc6a91062abb09e28e1370c78b015e717710343..3fb49721e90ebf8ed1268d9cb3590eae9d2ae358 100644
--- a/src/pystencils_autodiff/lbm/_autodiff_lbstep.py
+++ b/src/pystencils_autodiff/lbm/_autodiff_lbstep.py
@@ -4,7 +4,7 @@ import numpy as np
 
 import pystencils
 import pystencils_autodiff
-from lbmpy.autodiff.adjoint_boundaryconditions import AdjointBoundaryCondition
+from pystencils_autodiff.lbm.adjoint_boundaryconditions import AdjointBoundaryCondition
 from lbmpy.boundaries.boundaryhandling import LatticeBoltzmannBoundaryHandling
 from lbmpy.lbstep import LatticeBoltzmannStep
 from lbmpy.methods.abstractlbmethod import AbstractLbMethod, LbmCollisionRule