Need to transform test data too