Fix out of date docstring