Minor fix to dataset_config