diff --git a/source/ubinstall-gtk-log.c b/source/ubinstall-gtk-log.c
index fecb9a6..f0bd3a6 100644
--- a/source/ubinstall-gtk-log.c
+++ b/source/ubinstall-gtk-log.c
@@ -2,10 +2,9 @@
 
 
 void on_log_closed(GtkWidget *, log_window *window){
-    main_window *widgets = g_object_get_data(G_OBJECT(window->Window),"widgets");
 
-    gtk_widget_set_sensitive(widgets->ReadFullLogButton,1);
-    gtk_widget_set_sensitive(widgets->ReadShortLogButton,1);
+    GtkWidget *button = g_object_get_data(G_OBJECT(window->Window),"button");
+    gtk_widget_set_sensitive(button,1);
 
     free(window->command);
     window->Window=NULL;
@@ -87,6 +86,7 @@ void yon_read_log(GFileMonitor *,GFile *,GFile *,GFileMonitorEvent ,log_window *
 void on_process_log_view(GtkWidget *,main_window *widgets){
     log_window *window = yon_log_window_new();
     g_object_set_data(G_OBJECT(window->Window),"widgets",widgets);
+    g_object_set_data(G_OBJECT(window->Window),"button",widgets->ReadFullLogButton);
     g_signal_connect(G_OBJECT(window->Window),"destroy",G_CALLBACK(on_log_closed),window);
     gtk_widget_set_sensitive(widgets->ReadFullLogButton,0);
     yon_gtk_window_setup(GTK_WINDOW(window->Window),NULL,INSTALL_LOG_LABEL,icon_path,"log_viewer");
@@ -101,6 +101,7 @@ void on_process_log_view(GtkWidget *,main_window *widgets){
 void on_summary_log_view(GtkWidget *,main_window *widgets){
     log_window *window = yon_log_window_new();
     g_object_set_data(G_OBJECT(window->Window),"widgets",widgets);
+    g_object_set_data(G_OBJECT(window->Window),"button",widgets->ReadShortLogButton);
     g_signal_connect(G_OBJECT(window->Window),"destroy",G_CALLBACK(on_log_closed),window);
     gtk_widget_set_sensitive(widgets->ReadShortLogButton,0);
     yon_gtk_window_setup(GTK_WINDOW(window->Window),NULL,PROGRESS_LOG_LABEL,icon_path,"log_viewer");