Update comment to match new label