use smaller test dataset