Fix comment location