diff --git a/doc/notebooks/demo_thermalized_lbm.ipynb b/doc/notebooks/demo_thermalized_lbm.ipynb
index 9c0d957919c1539ff88250d354364c47160355a0..b9f759bd679c8a45e39ba5b662d6cc06185d8bbc 100644
--- a/doc/notebooks/demo_thermalized_lbm.ipynb
+++ b/doc/notebooks/demo_thermalized_lbm.ipynb
@@ -183,7 +183,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -216,7 +216,7 @@
     }
    ],
    "source": [
-    "from pystencils.sympyextensions import CastFunc\n",
+    "from pystencils.sympyextensions.typed_sympy import CastFunc\n",
     "\n",
     "dh = ps.create_data_handling(domain_size=(80, 80))\n",
     "\n",
@@ -347,11 +347,8 @@
   }
  ],
  "metadata": {
-  "interpreter": {
-   "hash": "ca06c80c4febc35b85e85156d391051f9f4a8895eee3f708eb1f33a09d8697a0"
-  },
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
diff --git a/src/lbmpy/boundaries/boundaryconditions.py b/src/lbmpy/boundaries/boundaryconditions.py
index 37304460d44bbfb89de12ed2f501d2453af39cb7..3021450de2693b15a53a64073ec0f1e61c3bc38c 100644
--- a/src/lbmpy/boundaries/boundaryconditions.py
+++ b/src/lbmpy/boundaries/boundaryconditions.py
@@ -3,7 +3,7 @@ from enum import Enum, auto
 from warnings import warn
 
 from pystencils import Assignment, AssignmentCollection, Field, TypedSymbol
-from pystencils.sympyextensions import CastFunc
+from pystencils.sympyextensions.typed_sympy import CastFunc
 from pystencils.stencil import offset_to_direction_string, direction_string_to_offset, inverse_direction
 from pystencils.sympyextensions import get_symmetric_part, simplify_by_equality, scalar_product
 
diff --git a/src/lbmpy/phasefield_allen_cahn/contact_angle.py b/src/lbmpy/phasefield_allen_cahn/contact_angle.py
index ef6131943e111c7e3b76097754be94f6f505082d..f7ed4b35f8a940037dd4bb83e099864dc9bf8b4f 100644
--- a/src/lbmpy/phasefield_allen_cahn/contact_angle.py
+++ b/src/lbmpy/phasefield_allen_cahn/contact_angle.py
@@ -12,7 +12,7 @@ from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo
 from pystencils.boundaries.boundaryconditions import Boundary
 
 from pystencils import TypedSymbol
-from pystencils.sympyextensions import CastFunc
+from pystencils.sympyextensions.typed_sympy import CastFunc
 
 
 class ContactAngle(Boundary):