Yes exactly like you described!
I just guess the game could have made it clearer, that if I didn't choose to forget a spell, I won't get the other one. I just kept the "forget slot" empty and there was still a checkmark on that screen like it was a valid choice.