diff --git a/docker/spleeter-conda.dockerfile b/docker/spleeter-conda.dockerfile
index 2a21cd8..9eb2883 100644
--- a/docker/spleeter-conda.dockerfile
+++ b/docker/spleeter-conda.dockerfile
@@ -2,12 +2,11 @@ ARG BASE=conda
 
 FROM ${BASE}
 
-ARG SPLEETER_PACKAGE=spleeter
 ARG SPLEETER_VERSION=1.5.3
 ENV MODEL_PATH /model
 
 RUN mkdir -p /model
 RUN conda install -y -c conda-forge musdb
-RUN conda install -y -c conda-forge ${SPLEETER_PACKAGE}==${SPLEETER_VERSION}
+RUN conda install -y -c conda-forge spleeter==${SPLEETER_VERSION}
 COPY docker/conda-entrypoint.sh spleeter-entrypoint.sh
 ENTRYPOINT ["/bin/bash", "spleeter-entrypoint.sh"]
\ No newline at end of file
diff --git a/docker/spleeter.dockerfile b/docker/spleeter.dockerfile
index 1161c3a..745ceae 100644
--- a/docker/spleeter.dockerfile
+++ b/docker/spleeter.dockerfile
@@ -2,13 +2,12 @@ ARG BASE=python:3.6
 
 FROM ${BASE}
 
-ARG SPLEETER_PACKAGE=spleeter
 ARG SPLEETER_VERSION=1.5.3
 ENV MODEL_PATH /model
 
 RUN mkdir -p /model
 RUN apt-get update && apt-get install -y ffmpeg libsndfile1
 RUN pip install musdb museval
-RUN pip install ${SPLEETER_PACKAGE}==${SPLEETER_VERSION}
+RUN pip install spleeter==${SPLEETER_VERSION}
 
 ENTRYPOINT ["spleeter"]