diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java
index aa04285..1e700a0 100644
--- a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java
+++ b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java
@@ -68,14 +68,19 @@ public static void build(Tensor<?> tensor, String memoryName) throws IllegalArgu
         {
             case UINT8:
             	buildFromTensorUByte((Tensor<UInt8>) tensor, memoryName);
+            	break;
             case INT32:
             	buildFromTensorInt((Tensor<Integer>) tensor, memoryName);
+            	break;
             case FLOAT:
             	buildFromTensorFloat((Tensor<Float>) tensor, memoryName);
+            	break;
             case DOUBLE:
             	buildFromTensorDouble((Tensor<Double>) tensor, memoryName);
+            	break;
             case INT64:
             	buildFromTensorLong((Tensor<Long>) tensor, memoryName);
+            	break;
             default:
                 throw new IllegalArgumentException("Unsupported tensor type: " + tensor.dataType().name());
         }