Fix return types in docstrings