diff --git a/QgisModelBaker/gui/panel/session_panel.py b/QgisModelBaker/gui/panel/session_panel.py
index a6be71e43..797bf8c12 100644
--- a/QgisModelBaker/gui/panel/session_panel.py
+++ b/QgisModelBaker/gui/panel/session_panel.py
@@ -288,12 +288,17 @@ def run(self, edited_command=None):
                     self.progress_bar.setValue(0)
                     if not self.db_action_type == DbActionType.GENERATE:
                         self.set_button_to_create_without_constraints()
+                    else:
+                        self.set_button_to_create()
                     self.is_running = False
                     return False
             except JavaNotFoundError as e:
                 self.print_info.emit(e.error_string, LogColor.COLOR_FAIL)
                 self.progress_bar.setValue(0)
-                self.set_button_to_create_without_constraints()
+                if not self.db_action_type == DbActionType.GENERATE:
+                    self.set_button_to_create_without_constraints()
+                else:
+                    self.set_button_to_create()
                 self.is_running = False
                 return False