diff --git a/src/main/java/ai/nets/samj/gui/MainGUI.java b/src/main/java/ai/nets/samj/gui/MainGUI.java
index c12eaf0..79b38aa 100644
--- a/src/main/java/ai/nets/samj/gui/MainGUI.java
+++ b/src/main/java/ai/nets/samj/gui/MainGUI.java
@@ -524,6 +524,7 @@ private < T extends RealType< T > & NativeType< T > > void batchSAMize() {
     	}
     	lyt.show(cardPanel2_2, INVISIBLE_STR);
     	this.stopProgressBtn.setEnabled(true);
+    	consumer.setFocusedImage(this.cmbImages.getSelectedObject());
     	new Thread(() -> {
     		try {
 				cmbModels.getSelectedModel().processBatchOfPrompts(pointPrompts, rectPrompts, rai, batchDrawerCallback);