Cleanups

1 files changed, 2 insertions, 4 deletions

diff --git a/hardinfo2/shell.c b/hardinfo2/shell.c
index 420d690e..bbe14dbb 100644
--- a/hardinfo2/shell.c
+++ b/hardinfo2/shell.c
@@ -674,7 +674,7 @@ static gboolean reload_section(gpointer data)
 	/* if there was a selection, reselect it */
 	if (path) {
 	    gtk_tree_selection_select_path(shell->info->selection, path);
-            gtk_tree_view_set_cursor(GTK_TREE_VIEW(shell->tree->view), path, NULL,
+            gtk_tree_view_set_cursor(GTK_TREE_VIEW(shell->info->view), path, NULL,
                                      FALSE);
 	    gtk_tree_path_free(path);
 	}
@@ -738,9 +738,6 @@ static void set_view_type(ShellViewType viewtype)
     gtk_tree_view_column_set_visible(shell->info->col_progress, FALSE);
     gtk_tree_view_column_set_visible(shell->info->col_value, TRUE);
     
-    /* turn off the column headers */
-    gtk_tree_view_set_headers_visible(GTK_TREE_VIEW(shell->info->view), FALSE);
-    
     /* turn off the rules hint */
     gtk_tree_view_set_rules_hint(GTK_TREE_VIEW(shell->info->view), FALSE);
 
@@ -751,6 +748,7 @@ static void set_view_type(ShellViewType viewtype)
     default:
     case SHELL_VIEW_NORMAL:
 	gtk_widget_hide(shell->notebook);
+        gtk_tree_view_set_headers_visible(GTK_TREE_VIEW(shell->info->view), FALSE);
 	break;
     case SHELL_VIEW_DUAL:
 	gtk_notebook_set_page(GTK_NOTEBOOK(shell->notebook), 0);