Added missing variable check

Also updated comment wording a bit.