Fix missing parameter from docstring