Fix docstring for sphinx