Fix answer checking