diff --git a/docker/Dockerfile b/docker/Dockerfile
index a54f1b45ca0..f08f5f035e7 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -138,12 +138,12 @@ COPY --from=build $VENV $VENV
 COPY --from=build sparseml sparseml
 ENV PATH="${VENV}/bin:$PATH"
 HEALTHCHECK CMD python -c 'import sparseml'
-ENTRYPOINT bash
+CMD bash
 
 FROM base as prod
 ARG VENV
 COPY --from=build $VENV $VENV
 ENV PATH="${VENV}/bin:$PATH"
 HEALTHCHECK CMD python -c 'import sparseml'
-ENTRYPOINT bash
+CMD bash