adding missing returnfig arg

py4dstem · Dec 11, 2023 · 31fb037 · 31fb037
1 parent b3e90e2
commit 31fb037
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py4DSTEM/braggvectors/braggvectors.py b/py4DSTEM/braggvectors/braggvectors.py
@@ -412,7 +412,7 @@ def plot(
         from py4DSTEM.process.diffraction import plot_diffraction_pattern
 
         if returnfig:
-            return plot_diffraction_pattern(self, **kwargs)
+            return plot_diffraction_pattern(self, returnfig=returnfig, **kwargs)
         else:
             plot_diffraction_pattern(self, **kwargs)