diff --git a/Dockerfile b/Dockerfile
index 27799d4..f309da4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,9 +8,8 @@ RUN apk add --no-cache curl && \
 
 ENV PATH="/root/.cargo/bin:$PATH"
 
-COPY Cargo.toml /Cargo.toml
-COPY main.rs /main.rs
-RUN cd / && cargo build --release && mkdir -p /app && cp /target/release/ssh-tui /app/tui
+COPY . /app
+RUN cd /app && cargo build --release && cp /target/release/ssh-tui /app/tui
 
 EXPOSE 22
 CMD ["/usr/local/bin/sshserver"]