diff --git a/nobrainer/models/labels_to_image_model.py b/nobrainer/models/labels_to_image_model.py
index cdc46033..ebb10ab5 100644
--- a/nobrainer/models/labels_to_image_model.py
+++ b/nobrainer/models/labels_to_image_model.py
@@ -13,10 +13,6 @@
 License.
 """
 
-# 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
 import keras.layers as KL
 from keras.models import Model
 
@@ -24,6 +20,11 @@
 import numpy as np
 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
+
 
 def labels_to_image_model(
     labels_shape,