diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 62de5a1..88f28a5 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -2,6 +2,7 @@
 FROM ubuntu:26.04 AS builder
 
 ENV DEBIAN_FRONTEND=noninteractive
+ENV REBUILD_FLAG=1
 
 RUN apt update && apt upgrade -y && apt install -y \
     build-essential \