diff --git a/pystencils/fd/spatial.py b/pystencils/fd/spatial.py
index e3ef030d1fa64ac619f2a8acedc8504e6c283638..a4e0974a53f4a15462eab3fc698785a11dae5393 100644
--- a/pystencils/fd/spatial.py
+++ b/pystencils/fd/spatial.py
@@ -1,10 +1,8 @@
 from typing import Tuple
 import sympy as sp
-from functools import partial
-
 from pystencils.astnodes import LoopOverCoordinate
 from pystencils.cache import memorycache
-from pystencils import AssignmentCollection, Field
+from pystencils import Field
 from pystencils.fd import Diff
 from pystencils.transformations import generic_visit
 from .derivative import diff_args
@@ -164,8 +162,10 @@ def discretize_spatial_staggered(expr, dx, stencil=fd_stencils_standard):
 
     return generic_visit(expr, visitor)
 
+
 # -------------------------------------- special stencils --------------------------------------------------------------
 
+
 @memorycache(maxsize=1)
 def forth_order_2d_derivation() -> Tuple[FiniteDifferenceStencilDerivation.Result, ...]:
     # Symmetry, isotropy and 4th order conditions are not enough to fully specify the stencil