Update to new test dataset