diff --git a/dfs-diffusers/docker/gpu/Dockerfile b/dfs-diffusers/docker/gpu/Dockerfile
index b7bf86a..bb8a63f 100644
--- a/dfs-diffusers/docker/gpu/Dockerfile
+++ b/dfs-diffusers/docker/gpu/Dockerfile
@@ -11,7 +11,7 @@ RUN pip3 install --no-cache-dir -r requirements.txt
 
 COPY download.py .
 
-RUN python3 download.py --model $MODEL_ID --refiner $REFINER_ID
+RUN python3 download.py --model $MODEL_ID --refiner "$REFINER_ID"
 
 COPY . .
 
diff --git a/dfs-diffusers/download.py b/dfs-diffusers/download.py
index 98df28e..de68de4 100644
--- a/dfs-diffusers/download.py
+++ b/dfs-diffusers/download.py
@@ -10,7 +10,7 @@
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--model", help="Model to download")
-parser.add_argument("--refiner", required=False, help="Refiner model to download")
+parser.add_argument("--refiner", help="Refiner model to download")
 args = parser.parse_args()
 
 print(f"Downloading model {args.model}")