diff --git a/Dockerfile b/Dockerfile
index 262dfa0..b7948d1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -25,7 +25,7 @@ RUN chmod +r ./Customers.csv
 RUN chmod +x ./train.py
 #RUN chmod +x ./test.py
 
-ARG epochs = 5
+ARG epochs=5
 RUN echo $epochs
 
 CMD python3 ./train.py