fix docstring tests