diff --git a/doc/screenshots/IJ-Menu.png b/doc/screenshots/IJ-Menu.png
index 62e2968b..d4b99dfe 100644
Binary files a/doc/screenshots/IJ-Menu.png and b/doc/screenshots/IJ-Menu.png differ
diff --git a/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java b/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java
index 27023c61..18034c9f 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