Make conditional more flexible and add a comment.