Removed wrong comment and made if condition simpler