diff --git a/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java b/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java
index a8b95648c1cf46277c82c94ac1805041610a792d..3332be6996af203c5d60fb7dfe1fd66190597a12 100644
--- a/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java
+++ b/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java
@@ -31,7 +31,7 @@ public final class TensorBuilder
      */
     public static Tensor<?> build(org.bioimageanalysis.icy.deeplearning.tensor.Tensor tensor)
     {
-    	return build(tensor.getDataAsNDArray());
+    	return build(tensor.getData());
     }
 
     /**
diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index c9ae2aba10d0931aab48b7c31b6f753b0154eb06..ec4315862462321d6884f8ebc8f224d17ac7a574 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -120,7 +120,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
 		if (outputNDArrays.size() != outputTensors.size())
 			throw new RunModelException(outputNDArrays.size(), outputTensors.size());
 		for (int i = 0; i < outputNDArrays.size(); i ++) {
-			outputTensors.get(i).setNDArrayData(Nd4fBuilder.build(outputNDArrays.get(i)));
+			outputTensors.get(i).setData(Nd4fBuilder.build(outputNDArrays.get(i)));
 		}
 		return outputTensors;
 	}