diff --git a/code/compute_results.py b/code/compute_results.py
index e9c93e18f5d804a39ad98c47893850173ef17407..40e137133f914d40f886a0c15ddcaa8f5c66b89c 100644
--- a/code/compute_results.py
+++ b/code/compute_results.py
@@ -121,7 +121,7 @@ if __name__ == "__main__":
     parser.add_argument('--stage', nargs='?', type=int, required=True, help='Specify the stage number among [1, 4].')
     parser.add_argument('--experiment_ids', nargs='+', type=int, required=True, help='Compute the results of the specified experiment id(s).' + \
         'stage=1: {{base_with_params}} {{random_with_params}} {{omp_with_params}} {{base_wo_params}} {{random_wo_params}} {{omp_wo_params}}')
-    parser.add_argument('--dataset_name', nargs='?', type=str, help='Specify the dataset name. TODO: read it from models dir directly.')
+    parser.add_argument('--dataset_name', nargs='?', type=str, required=True, help='Specify the dataset name. TODO: read it from models dir directly.')
     parser.add_argument('--results_dir', nargs='?', type=str, default=DEFAULT_RESULTS_DIR, help='The output directory of the results.')
     parser.add_argument('--models_dir', nargs='?', type=str, default=DEFAULT_MODELS_DIR, help='The output directory of the trained models.')
     args = parser.parse_args()