Also fixing dataset download