Update comment on @return