Update comment and return type [ci skip]