Fixed the docstring tests.