one more bug fix

py4dstem · Nov 2, 2023 · a51594c · a51594c
1 parent 3a6ee5a
commit a51594c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py4DSTEM/process/phase/iterative_multislice_ptychography.py b/py4DSTEM/process/phase/iterative_multislice_ptychography.py
@@ -3004,7 +3004,7 @@ def show_slices(
         cmap = kwargs.pop("cmap", "magma")
 
         if common_color_scale:
-            vals = np.sort(rotated_object.mean(0).ravel())
+            vals = np.sort(rotated_object.ravel())
             ind_vmin = np.round((vals.shape[0] - 1) * 0.02).astype("int")
             ind_vmax = np.round((vals.shape[0] - 1) * 0.98).astype("int")
             ind_vmin = np.max([0, ind_vmin])