Correct some docstrings to reflect the updated API.