diff --git a/lbmpy/session.py b/lbmpy/session.py
index 8c9b2ac3a8cdfb1ff33c32645847c402a01ea90e..cb229ba3520cf488865233a806ea03dcbc166b4f 100644
--- a/lbmpy/session.py
+++ b/lbmpy/session.py
@@ -1,4 +1,5 @@
 from pystencils.sympy_gmpy_bug_workaround import *
+import pystencils as ps
 import sympy as sp
 import numpy as np
 from lbmpy.scenarios import *
@@ -10,4 +11,4 @@ from lbmpy.lbstep import LatticeBoltzmannStep
 from lbmpy.geometry import *
 from lbmpy.parameterization import ScalingWidget
 import lbmpy.plot2d as plt
-from pystencils.jupytersetup import *
\ No newline at end of file
+from pystencils.jupytersetup import *