diff --git a/source/ubinstall-gtk.c b/source/ubinstall-gtk.c
index caf4693..f34c5bf 100644
--- a/source/ubinstall-gtk.c
+++ b/source/ubinstall-gtk.c
@@ -897,6 +897,7 @@ void on_page_navigation_clicked(GtkWidget *self, main_window *widgets){
                     gtk_notebook_set_current_page(GTK_NOTEBOOK(widgets->Notebook),YON_PAGE_INSTALLATION);
                 } break;
 
+                case YON_PAGE_INSTALL_ERROR:
                 case YON_PAGE_COMPLETION:
                     if (!system("reboot"))
                     return;
@@ -1011,6 +1012,10 @@ void on_page_navigation_clicked(GtkWidget *self, main_window *widgets){
                     gtk_notebook_set_current_page(GTK_NOTEBOOK(widgets->Notebook),YON_PAGE_INSTALL_OPTIONS);
                     break;
 
+                case YON_PAGE_INSTALLATION_BEGIN:
+                    gtk_notebook_set_current_page(GTK_NOTEBOOK(widgets->Notebook),YON_PAGE_SECTIONS);
+                break; 
+
                 default:{
                     gtk_notebook_set_current_page(GTK_NOTEBOOK(widgets->Notebook),page-1);
                 }
@@ -1019,7 +1024,8 @@ void on_page_navigation_clicked(GtkWidget *self, main_window *widgets){
         } break;
         case -2: { // Cancel/repeat
             switch (page){
-                case YON_PAGE_COMPLETION:{
+                case YON_PAGE_COMPLETION:
+                case YON_PAGE_INSTALL_ERROR:{
                     gtk_main_quit();
                     }break;
                 default:{