diff --git a/recipes/recipe_modules/bot_update/tests/ensure_checkout.py b/recipes/recipe_modules/bot_update/tests/ensure_checkout.py
index a1fe200f89..c9043a57a1 100644
--- a/recipes/recipe_modules/bot_update/tests/ensure_checkout.py
+++ b/recipes/recipe_modules/bot_update/tests/ensure_checkout.py
@@ -19,7 +19,7 @@ def RunSteps(api):
 def GenTests(api):
   yield (
       api.test('basic') +
-      api.post_process(post_process.StatusCodeIn, 0) +
+      api.post_process(post_process.StatusSuccess) +
       api.post_process(post_process.DropExpectation)
   )
 
@@ -29,6 +29,6 @@ def GenTests(api):
           'bot_update',
           api.json.output({'did_run': True}),
           retcode=1) +
-      api.post_process(post_process.StatusCodeIn, 1) +
+      api.post_process(post_process.StatusAnyFailure) +
       api.post_process(post_process.DropExpectation)
   )