fix a docstring typo in optimize.py