diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index 8e8adf025ed0869b5a710e071ffe12a852b2f2e5..9e605ce208af1620c49d5734dc585bd0ea7b9cb3 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -3,20 +3,19 @@ package tensorflow.v1;
 import java.util.ArrayList;
 import java.util.List;
 
-import org.bioimageanalysis.icy.deeplearning.tensor.Tensor;
 import org.bioimageanalysis.icy.deeplearning.exceptions.LoadModelException;
 import org.bioimageanalysis.icy.deeplearning.exceptions.RunModelException;
+import org.bioimageanalysis.icy.deeplearning.tensor.Tensor;
 import org.bioimageanalysis.icy.deeplearning.utils.DeepLearningInterface;
 import org.bioimageanalysis.icy.tensorflow.v1.tensor.Nd4fBuilder;
 import org.bioimageanalysis.icy.tensorflow.v1.tensor.TensorBuilder;
+import org.nd4j.shade.protobuf.InvalidProtocolBufferException;
 import org.tensorflow.SavedModelBundle;
 import org.tensorflow.Session;
 import org.tensorflow.framework.MetaGraphDef;
 import org.tensorflow.framework.SignatureDef;
 import org.tensorflow.framework.TensorInfo;
 
-import com.google.protobuf.InvalidProtocolBufferException;
-
 
 /**
  * This plugin includes the libraries to convert back and forth TensorFlow 1 to Sequences and IcyBufferedImages.