Fix docstring

The previous docstring was resulting in None