diff --git a/disk/InteractionDisk_temp.py b/disk/InteractionDisk_temp.py
index 88e57d4abc43cfee74da428b6b6e8aa788c7c508..ee6f1d1040f694831f01adfda1098501dbeca82f 100644
--- a/disk/InteractionDisk_temp.py
+++ b/disk/InteractionDisk_temp.py
@@ -16,6 +16,7 @@ import skimage.io
 import neural_network as nn
 import pytiff
 import hungarian as hu
+from segment import segment
 
 
 class Reader:
@@ -437,7 +438,7 @@ class Reader:
                                                           self.tlabels[currentT])])
             pred = np.array(fileprediction['/{}/{}'.format(self.fovlabels[currentFOV], self.tlabels[currentT])])	
             fileprediction.close()
-            segmentedmask = nn.segment(tmpthrmask, pred, segparamvalue)	# SJR: added to read out the prediction as well
+            segmentedmask = segment(tmpthrmask, pred, segparamvalue)	# SJR: added to read out the prediction as well
             filethr.close()
             return segmentedmask