diff --git a/test/e2e/test-openshell-gateway-upgrade.sh b/test/e2e/test-openshell-gateway-upgrade.sh
index 1f243d9eaf..1408ca799c 100755
--- a/test/e2e/test-openshell-gateway-upgrade.sh
+++ b/test/e2e/test-openshell-gateway-upgrade.sh
@@ -135,6 +135,9 @@ rewrote_base=0
 while [ "$#" -gt 0 ]; do
   case "$1" in
     --build-arg)
+      if [ "$#" -ge 2 ] && [ "${2#BASE_IMAGE=}" != "$2" ]; then
+        rewrote_base=1
+      fi
       if [ "$#" -ge 2 ] && [ "${2#OPENCLAW_VERSION=}" != "$2" ]; then
         args+=("--build-arg" "OPENCLAW_VERSION=${old_openclaw}")
         rewrote_openclaw=1
@@ -164,6 +167,9 @@ while [ "$#" -gt 0 ]; do
       shift
       continue
       ;;
+    --build-arg=BASE_IMAGE=*)
+      rewrote_base=1
+      ;;
   esac
   args+=("$1")
   shift