Fix docstring for consistency