diff --git a/noxfile.py b/noxfile.py
index 9e723b9a47f5f7f8de09b79086219b5aa1d1cbe1..16dbc36e9f68df41547430cac6f9f0ea47aff599 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -76,7 +76,8 @@ def install_pystencils_master(session: nox.Session):
 
 
 def install_sympy_master(session: nox.Session):
-    session.install("git+https://i10git.cs.fau.de/pycodegen/pystencils.git@master")
+    session.install("--upgrade", "git+https://github.com/sympy/sympy.git@master")
+
 
 @nox.session(python="3.10", tags=["qa", "code-quality"])
 def lint(session: nox.Session):
diff --git a/src/lbmpy/advanced_streaming/communication.py b/src/lbmpy/advanced_streaming/communication.py
index c02b6e2229d3f1a3da0e81cfdd70a6e5c5ff5e87..6d5c46303676b5bf17af63fcb774408d14863604 100644
--- a/src/lbmpy/advanced_streaming/communication.py
+++ b/src/lbmpy/advanced_streaming/communication.py
@@ -1,5 +1,5 @@
 import itertools
-from pystencils import CreateKernelConfig, Field, Assignment, AssignmentCollection
+from pystencils import CreateKernelConfig, Field, Assignment, AssignmentCollection, Target
 from pystencils.slicing import (
     shift_slice,
     get_slice_before_ghost_layer,
@@ -14,7 +14,6 @@ from lbmpy.advanced_streaming.utility import (
     numeric_offsets,
 )
 from pystencils.datahandling import SerialDataHandling
-from pystencils.enums import Target
 from itertools import chain