diff --git a/Dockerfile b/Dockerfile
index 37d69a7..83cf00a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -21,4 +21,4 @@ COPY src ./src
 RUN mkdir -p ./data; mkdir -p ./results
 
 
-CMD kaggle datasets download -p data --unzip clmentbisaillon/fake-and-real-news-dataset && python ./src/main.py "--$RUN_TYPE"
\ No newline at end of file
+CMD kaggle datasets download -p data --unzip clmentbisaillon/fake-and-real-news-dataset && python ./dataset.py --dataset && python ./src/main.py "--$RUN_TYPE"
\ No newline at end of file