STY: Fix docstring