Setting the prediction set in uniform way for both body and head grammar slots.