diff --git a/Dockerfile b/Dockerfile
index 587ab9b8..73a009ae 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -211,7 +211,7 @@ RUN chmod +x entrypoint.sh
 ENTRYPOINT ["./entrypoint.sh"]
 
 # Final image
-FROM base
+FROM base as tgi
 
 ENTRYPOINT ["text-generation-launcher"]
 CMD ["--json-output"]