diff --git a/py4DSTEM/process/diffraction/crystal_viz.py b/py4DSTEM/process/diffraction/crystal_viz.py
index 3318215ae..4e86c2524 100644
--- a/py4DSTEM/process/diffraction/crystal_viz.py
+++ b/py4DSTEM/process/diffraction/crystal_viz.py
@@ -3,8 +3,8 @@
 from mpl_toolkits.mplot3d import art3d
 from scipy.signal import medfilt
 from scipy.ndimage import gaussian_filter
-from scipy.ndimage.morphology import distance_transform_edt
-from skimage.morphology import erosion
+from scipy.ndimage import distance_transform_edt
+from skimage.morphology import dilation, erosion
 
 import warnings
 import numpy as np