Updated comment, fixed if check