Update comment accordingly