Reduce test set for now
Reduce test set for now