fix docstring so it passes test