Update comment to reflect change in layout