Fix the check answer function