Update comment, add back removed line