diff --git a/Dockerfile b/Dockerfile
index c4fcb35..8783391 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,5 +6,6 @@ RUN CGO_ENABLED=0 GOOS=linux go build -mod=vendor -o go-websocket-chat-demo .
 
 FROM heroku/heroku:18
 WORKDIR /app
-COPY --from=0 /src/go-websocket-chat-demo /app
-CMD ["./go-websocket-chat-demo"]
\ No newline at end of file
+COPY ./public /app/public
+COPY --from=0 /src/go-websocket-chat-demo /app/
+CMD ["/app/go-websocket-chat-demo"]
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
index 4a355a3..93f02ae 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -2,7 +2,6 @@ version: '2'
 
 services:
   app:
-    command: go-websocket-chat-demo
     build: .
     environment:
       PORT: '5000'