I don’t know which part’s correct in this one, but the grammar point uses に only while the answer (and the sentence in the examples) uses の. Not sure if one or both particles are correct here (I don’t see why it couldn’t be に), but either way there’s definitely some confusion.