fix + docstring