diff --git a/src/main/java/ai/nets/samj/gui/MainGUI.java b/src/main/java/ai/nets/samj/gui/MainGUI.java
index 9602bff..404d4bb 100644
--- a/src/main/java/ai/nets/samj/gui/MainGUI.java
+++ b/src/main/java/ai/nets/samj/gui/MainGUI.java
@@ -103,7 +103,6 @@ public MainGUI(List<SAMModel> modelList, ConsumerInterface consumer) {
         createListeners();
         this.consumer = consumer;
         this.consumer.setCallback(consumerCallback);
-        consumerCallback.validPromptChosen(consumer.isValidPromptSelected());
         cmbImages = ImageSelectionOnlyComboBox.create(this.consumer, imageListener);
 
         if (modelList == null) this.modelList = DEFAULT_MODEL_LIST;
@@ -160,6 +159,8 @@ public MainGUI(List<SAMModel> modelList, ConsumerInterface consumer) {
 
         // Set the initial size of the frame
         setSize(MAIN_HORIZONTAL_SIZE, MAIN_VERTICAL_SIZE); // Width x Height
+        
+        consumerCallback.validPromptChosen(consumer.isValidPromptSelected());
 
         this.addWindowListener(new WindowAdapter() {
             @Override