DOC: Fixing formula in docstrings

The formula in the docstring did not match the formula in the implementation.