After applying to test set