Updated comment following PR comments