Fix docstring parsing