diff --git a/backend/Dockerfile b/backend/Dockerfile
index 31c70e4..5dfc713 100644
--- a/backend/Dockerfile
+++ b/backend/Dockerfile
@@ -13,7 +13,8 @@
 #   docker build -t nat20-notes-backend .
 # ==========================================
 
-FROM nat20-whisperx-base:latest
+ARG BASE_IMAGE=nat20-whisperx-base:latest
+FROM $BASE_IMAGE AS runtime
 
 WORKDIR /app