docs: Fix docstring