Update comment to reflect changes to logic