diff --git a/nobrainer/models/labels_to_image_model.py b/nobrainer/models/labels_to_image_model.py
index ebb10ab5..70ae0c5b 100644
--- a/nobrainer/models/labels_to_image_model.py
+++ b/nobrainer/models/labels_to_image_model.py
@@ -21,9 +21,9 @@
 import tensorflow as tf
 
 # third-party imports
-from nobrainer.SynthSeg.ext.lab2im import edit_tensors as l2i_et
-from nobrainer.SynthSeg.ext.lab2im import layers, utils
-from nobrainer.SynthSeg.ext.lab2im.edit_volumes import get_ras_axes
+from nobrainer.ext.SynthSeg.lab2im import edit_tensors as l2i_et
+from nobrainer.ext.SynthSeg.lab2im import layers, utils
+from nobrainer.ext.SynthSeg.lab2im.edit_volumes import get_ras_axes
 
 
 def labels_to_image_model(