diff --git a/cellacdc/dataPrep.py b/cellacdc/dataPrep.py
index 8c751ca3..65df0188 100755
--- a/cellacdc/dataPrep.py
+++ b/cellacdc/dataPrep.py
@@ -12,7 +12,6 @@
 import scipy.interpolate
 import skimage
 import skimage.io
-from torch import int16
 from tqdm import tqdm
 from functools import partial, wraps
 from tifffile.tifffile import TiffWriter, TiffFile