Fixed misplaced comments