diff --git a/comfyui-cuda-dockerfile b/comfyui-cuda-dockerfile
index 49340f6..263a39a 100644
--- a/comfyui-cuda-dockerfile
+++ b/comfyui-cuda-dockerfile
@@ -7,6 +7,8 @@ RUN pip install imageio controlnet_aux
 # Set working directory
 WORKDIR /app
 
+ARG gitversion
+
 # Clone ComfyUI
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git .