Download fixed for custom datasets