diff --git a/postinstall_runner_action.cc b/postinstall_runner_action.cc
index 20989aa..7ee4143 100644
--- a/postinstall_runner_action.cc
+++ b/postinstall_runner_action.cc
@@ -67,7 +67,7 @@
   ScopedTempUnmounter temp_unmounter(temp_rootfs_dir_);
   if (return_code != 0) {
     LOG(ERROR) << "Postinst command failed with code: " << return_code;
-    if (return_code == 2) {
+    if (return_code == 3) {
       // 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.
diff --git a/postinstall_runner_action_unittest.cc b/postinstall_runner_action_unittest.cc
index 2db4edc..69a5eb0 100644
--- a/postinstall_runner_action_unittest.cc
+++ b/postinstall_runner_action_unittest.cc
@@ -75,7 +75,7 @@
 
 TEST_F(PostinstallRunnerActionTest, RunAsRootFirmwareBErrScriptTest) {
   ASSERT_EQ(0, getuid());
-  DoTest(true, 2);
+  DoTest(true, 3);
 }
 
 void PostinstallRunnerActionTest::DoTest(bool do_losetup, int err_code) {