diff --git a/omaha_request_action_unittest.cc b/omaha_request_action_unittest.cc
index 66fc6fe..a642b5a 100644
--- a/omaha_request_action_unittest.cc
+++ b/omaha_request_action_unittest.cc
@@ -3187,6 +3187,7 @@
 
 TEST_F(OmahaRequestActionTest, PastRollbackVersionsValidEntries) {
   OmahaResponse response;
+  request_params_.set_rollback_allowed_milestones(4);
   fake_update_response_.rollback = true;
   fake_update_response_.rollback_allowed_milestones = 4;
   fake_update_response_.rollback_firmware_version = "4.3";