Fix test dataset