Fixing docstring for test-doc pass