diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index bb68bb70adb83bbd56379f1965fbe4cd11d4757a..47300d5508b38b062db49bcd28906944f3928933 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -78,7 +78,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
 	}
 
 	@Override
-	public List<Tensor> run(List<Tensor> inputTensors, List<Tensor> outputTensors) throws RunModelException {
+	public List<Tensor<?>> run(List<Tensor<?>> inputTensors, List<Tensor<?>> outputTensors) throws RunModelException {
 		Session session = model.session();
 		Session.Runner runner = session.runner();
         List<String> inputListNames = new ArrayList<String>();
@@ -117,7 +117,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
 	 * @throws RunModelException If the number of tensors expected is not the same as the number of
 	 * 	Tensors outputed by the model
 	 */
-	public static List<Tensor> fillOutputTensors(List<org.tensorflow.Tensor<?>> outputNDArrays, List<Tensor> outputTensors) throws RunModelException{
+	public static List<Tensor<?>> fillOutputTensors(List<org.tensorflow.Tensor<?>> outputNDArrays, List<Tensor<?>> outputTensors) throws RunModelException{
 		if (outputNDArrays.size() != outputTensors.size())
 			throw new RunModelException(outputNDArrays.size(), outputTensors.size());
 		for (int i = 0; i < outputNDArrays.size(); i ++) {