diff --git a/update_attempter.cc b/update_attempter.cc
index 81f423d..cda244d 100644
--- a/update_attempter.cc
+++ b/update_attempter.cc
@@ -1145,6 +1145,11 @@
           ret_value = false;
       }
 
+      // Update the boot flags so the current slot has higher priority.
+      BootControlInterface* boot_control = system_state_->boot_control();
+      if (!boot_control->SetActiveBootSlot(boot_control->GetCurrentSlot()))
+        ret_value = false;
+
       // Notify the PayloadState that the successful payload was canceled.
       system_state_->payload_state()->ResetUpdateStatus();