Fix test docstring