diff --git a/Dockerfile b/Dockerfile
index 1806b6b..775cde6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,6 +8,7 @@ COPY . .
 RUN swift build --enable-test-discovery -c release -Xswiftc -g
 
 # Run image
-FROM swift:5.10
+FROM swift:5.10-slim
 WORKDIR /run
-COPY --from=build /build/.build/release /run
+COPY --from=build /build/.build/release/dewPoint-controller /run
+CMD ["/bin/bash", "-xc", "./dewPoint-controller"]
diff --git a/docker-compose.yaml b/docker-compose.yaml
index 60a308f..54ff6a2 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -2,8 +2,8 @@
 services:
   server:
     image: swift-mqtt-dewpoint:latest
+    restart: unless-stopped
     env_file: .env
-    command: /bin/bash -xc "./dewPoint-controller"
 
   test:
     image: swift:latest