diff --git a/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java b/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java
index 27023c6..18034c9 100644
--- a/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java
+++ b/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java
@@ -46,13 +46,7 @@ public final class ObjectClassificationCommand<T extends NativeType<T> & RealTyp
 
     @Override
     protected List<String> workflowArgs() {
-        if (ROLE_PROBABILITIES.equals(secondInputType)) {
-            return Collections.singletonList("--export_source=Object Probabilities");
-        }
-        if (ROLE_SEGMENTATION.equals(secondInputType)) {
-            return Collections.singletonList("--export_source=Object Predictions");
-        }
-        throw new IllegalStateException("Unexpected value: " + secondInputType);
+        return Collections.singletonList("--export_source=Object Predictions");
     }
 
     @Override