diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/tensor/TensorBuilder.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/tensor/TensorBuilder.java
index e1724e1..0d0b5c3 100644
--- a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/tensor/TensorBuilder.java
+++ b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/tensor/TensorBuilder.java
@@ -99,7 +99,6 @@ public static < T extends RealType< T > & NativeType< T > >  TType build(io.bioi
 	 * @throws IllegalArgumentException if the type of the {@link RandomAccessibleInterval}
 	 *  is not supported
 	 */
-	@SuppressWarnings("unchecked")
 	public static < T extends RealType< T > & NativeType< T > > TType build(
 		RandomAccessibleInterval<T> array) throws IllegalArgumentException
 	{