Fixed import of other modality Segmentation output