diff --git a/dsms-gateway/Dockerfile b/dsms-gateway/Dockerfile
index 5ac37dc..b04598d 100644
--- a/dsms-gateway/Dockerfile
+++ b/dsms-gateway/Dockerfile
@@ -14,7 +14,7 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application
-COPY main.py .
+COPY . .
 
 # Environment variables
 ENV IPFS_API_URL=http://dsms-node:5001