Tweak docstring in test