Fixed missing arg in docstring