Forgot to update docstring