Fix docstring test failure