diff --git a/postinstall_runner_action.cc b/postinstall_runner_action.cc
index 80c28c9..2c8b943 100644
--- a/postinstall_runner_action.cc
+++ b/postinstall_runner_action.cc
@@ -65,6 +65,12 @@
   ScopedTempUnmounter temp_unmounter(temp_rootfs_dir_);
   if (return_code != 0) {
     LOG(ERROR) << "Postinst command failed with code: " << return_code;
+    if (return_code == 2) {
+      // This special return code means that we tried to update firmware,
+      // but couldn't because we booted from FW B, and we need to reboot
+      // to get back to FW A.
+      completer.set_code(kActionCodePostinstallBootedFromFirmwareB);
+    }
     return;
   }
   if (HasOutputPipe()) {