diff --git a/cmd/amlogic/cmd_avb.c b/cmd/amlogic/cmd_avb.c
index 948f7ed..df1b861 100644
--- a/cmd/amlogic/cmd_avb.c
+++ b/cmd/amlogic/cmd_avb.c
@@ -381,13 +381,10 @@
     printf("ab_suffix is %s\n", ab_suffix);
 
     AvbSlotVerifyFlags flags = AVB_SLOT_VERIFY_FLAGS_NONE;
-    char *upgradestep = NULL;
 
     avb_init();
 
-    upgradestep = env_get("upgrade_step");
-
-    if (is_device_unlocked() || !strcmp(upgradestep, "3"))
+    if (is_device_unlocked())
         flags |= AVB_SLOT_VERIFY_FLAGS_ALLOW_VERIFICATION_ERROR;
 
     if (!strcmp(ab_suffix, "")) {
@@ -412,9 +409,6 @@
             flags,
             AVB_HASHTREE_ERROR_MODE_RESTART_AND_INVALIDATE, out_data);
 
-    if (!strcmp(upgradestep, "3"))
-        result = AVB_SLOT_VERIFY_RESULT_OK;
-
     return result;
 }