Fix docstring tests.