diff --git a/docker/spleeter-model.dockerfile b/docker/spleeter-model.dockerfile
index 7bdffc4..5930f63 100644
--- a/docker/spleeter-model.dockerfile
+++ b/docker/spleeter-model.dockerfile
@@ -1,8 +1,8 @@
 ARG BASE=researchdeezer/spleeter
-ARG MODEL=2-stems
 
 FROM ${BASE}
 
+ARG MODEL=2-stems
 RUN mkdir -p /model/$MODEL \
     && wget -O /tmp/$MODEL.tar.gz https://github.com/deezer/spleeter/releases/download/v1.4.0/$MODEL.tar.gz \
     && tar -xvzf /tmp/$MODEL.tar.gz -C /model/$MODEL/ \