fix new_dataset test