Fix a comment and a docstring