diff --git a/py4DSTEM/process/strain/strain.py b/py4DSTEM/process/strain/strain.py
index 0b27d6562..c6b245192 100644
--- a/py4DSTEM/process/strain/strain.py
+++ b/py4DSTEM/process/strain/strain.py
@@ -1432,7 +1432,7 @@ def show_reference_directions(
             plt.show()
             return
         else:
-            return fig, (ax1,ax2)
+            return fig, (ax1, ax2)
 
     def show_lattice_vectors(
         ar,