diff --git a/scripts/download_dataset.py b/scripts/download_dataset.py
index 5713de5..70e822f 100755
--- a/scripts/download_dataset.py
+++ b/scripts/download_dataset.py
@@ -112,7 +112,7 @@ def download_file(filename, params):
         tmp_file = os.path.join(DATA_FOLDER, 'simulation', 'train.zip')
         params = {
             'path': '/simulation',
-            'files': 'test',
+            'files': 'train',
         }
         folder = os.path.join(DATA_FOLDER, 'simulation')
         approx_size = 152