diff --git a/source/ubinstall-gtk-installation.c b/source/ubinstall-gtk-installation.c
index 91d3466..580d7be 100644
--- a/source/ubinstall-gtk-installation.c
+++ b/source/ubinstall-gtk-installation.c
@@ -357,10 +357,7 @@ void on_device_selection_changed(GtkWidget *self, main_window *widgets){
 void yon_devices_setup(main_window *widgets){
     GtkTreeIter iter;
     gtk_list_store_clear(widgets->DevicesList);
-    gtk_list_store_clear(widgets->PartitionsList);
-    // int size;
-    // config_str parameters = NULL;
-    // parameters = yon_config_load(yon_debug_output("%s\n",get_devices_command),&size);
+    // gtk_list_store_clear(widgets->PartitionsList);
     for (int i=1;i<main_config.devices_size;i++){
         yon_char_remove_last_symbol(main_config.devices[i],'\n');
         int parsed_size;
@@ -490,6 +487,7 @@ void yon_install_init(main_window *widgets, enum YON_PAGES page){
             gtk_tree_model_get(model,&iter,0,&target,-1);
             if (!yon_char_is_empty(target)&&!strcmp(target,device)){
                 gtk_tree_selection_select_iter(gtk_tree_view_get_selection(GTK_TREE_VIEW(device_tree)),&iter);
+                on_device_selection_changed(device_tree,widgets);
                 break;
             }
         }
@@ -498,7 +496,7 @@ void yon_install_init(main_window *widgets, enum YON_PAGES page){
             for_iter(model,&iter){
                 char *target;
                 gtk_tree_model_get(model,&iter,0,&target,-1);
-                if (!yon_char_is_empty(target)&&!strcmp(target,device)){
+                if (!yon_char_is_empty(target)&&!strcmp(target,part)){
                     gtk_tree_selection_select_iter(gtk_tree_view_get_selection(GTK_TREE_VIEW(partition_tree)),&iter);
                     break;
                 }
@@ -560,7 +558,7 @@ void yon_install_init(main_window *widgets, enum YON_PAGES page){
                 gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(partition_size_combo),"G",_("Gb"));
                 gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(partition_size_combo),"T",_("Tb"));
                 gtk_combo_box_set_active(GTK_COMBO_BOX(partition_size_combo),0);
-                char *parameter = config(part_size_parameter);
+                char *parameter = yon_config_get(part_size_parameter);
                 if (!yon_char_is_empty(parameter)){
                     gtk_spin_button_set_value(GTK_SPIN_BUTTON(partition_size_spin),atol(parameter));
                     gtk_combo_box_set_active(GTK_COMBO_BOX(partition_size_combo),yon_get_size_get_from_letter(parameter[strlen(parameter)-1])-1);