Move comment inside the if