diff --git a/Dockerfile b/Dockerfile
index 662b5f7..9ecfaa1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -248,9 +248,12 @@ ENV SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=1
 ENV SYCL_CACHE_PERSISTENT=1
 ENV PATH=/opt/backstory:$PATH
 
+ENTRYPOINT [ "/entrypoint.sh" ]
+
+FROM backstory AS backstory-prod
+
 COPY /src/ /opt/backstory/src/
 
-ENTRYPOINT [ "/entrypoint.sh" ]
 
 FROM ubuntu:oracular AS ollama
 
diff --git a/docker-compose.yml b/docker-compose.yml
index f807c47..0d0255c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -39,7 +39,7 @@ services:
     build:
       context: .
       dockerfile: Dockerfile
-      target: backstory
+      target: backstory-prod
     #image: backstory
     container_name: backstory-prod
     restart: "always"