Change comment to match the code