Updated patch with docstring fix.