Fix comment to reflect previous changes