Fix checking correct answers logic