diff --git a/gui/action.cpp b/gui/action.cpp
index 819e4ad8..d5b8c8f4 100755
--- a/gui/action.cpp
+++ b/gui/action.cpp
@@ -1801,9 +1801,9 @@ int GUIAction::flashimage(std::string arg __unused)
 		string current_slot = PartitionManager.Get_Active_Slot_Display();
 		bool pre_op_status = PartitionManager.Flash_Image(path, filename);
 
-		PartitionManager.Set_Active_Slot(current_slot == "A" ? "B" : "A");
+		PartitionManager.Override_Active_Slot(current_slot == "A" ? "B" : "A");
 		op_status = (int) !(pre_op_status && PartitionManager.Flash_Image(path, filename));
-		PartitionManager.Set_Active_Slot(current_slot);
+		PartitionManager.Override_Active_Slot(current_slot);
 
 		DataManager::SetValue("tw_flash_both_slots", 0);
 		flag = false;
@@ -2360,4 +2360,4 @@ int GUIAction::mergesnapshots(string arg __unused) {
 	}
 	operation_end(op_status);
 	return 0;
-}
\ No newline at end of file
+}