diff --git a/docker-compose.base.yml b/docker-compose.base.yml
index f82e962..d4dfe81 100644
--- a/docker-compose.base.yml
+++ b/docker-compose.base.yml
@@ -27,14 +27,13 @@ services:
       context: .
       dockerfile: worker.Dockerfile
       args:
-        # FIXME
-        WHISPER_MODEL: small
+        WHISPER_MODEL: ${WHISPER_MODEL}
     networks:
       - app
     depends_on:
       - redis
     healthcheck:
-      test: ["CMD-SHELL", "celery -b redis://redis:6379/0 inspect ping -d celery@$$HOSTNAME"]
+      test: ["CMD-SHELL", "celery -b ${BROKER_URL} inspect ping -d celery@$$HOSTNAME"]
       interval: 5s
       timeout: 5s
       retries: 5
diff --git a/scripts/download_models.py b/scripts/download_models.py
index fb509a4..b7dd384 100644
--- a/scripts/download_models.py
+++ b/scripts/download_models.py
@@ -5,5 +5,3 @@ from whisper import _download, _MODELS  # type: ignore
 if __name__ == "__main__":
     model_name = sys.argv[1].strip()
     _download(_MODELS[model_name], "/models/", False)
-    # if model_name != "large":
-    #     _download(_MODELS[f"{model_name}.en"], "/models/", False)