diff --git a/localai-p2p-master/Dockerfile b/localai-p2p-master/Dockerfile
index dc98929..c181370 100644
--- a/localai-p2p-master/Dockerfile
+++ b/localai-p2p-master/Dockerfile
@@ -6,7 +6,6 @@ ENV LOCALAI_P2P="true"
 ENV LOCALAI_FEDERATED="true"
 
 
-COPY run.sh /run.sh
-RUN chmod +x /run.sh
-
-CMD [ "/run.sh" ]
+RUN mkdir -p /etc/services.d/localai
+COPY run.sh /etc/services.d/localai/run
+RUN chmod +x /etc/services.d/localai/run
diff --git a/localai-p2p-worker/Dockerfile b/localai-p2p-worker/Dockerfile
index dc98929..c181370 100644
--- a/localai-p2p-worker/Dockerfile
+++ b/localai-p2p-worker/Dockerfile
@@ -6,7 +6,6 @@ ENV LOCALAI_P2P="true"
 ENV LOCALAI_FEDERATED="true"
 
 
-COPY run.sh /run.sh
-RUN chmod +x /run.sh
-
-CMD [ "/run.sh" ]
+RUN mkdir -p /etc/services.d/localai
+COPY run.sh /etc/services.d/localai/run
+RUN chmod +x /etc/services.d/localai/run