Correct the check answer function