fix missing types in docstrings