Fix docs and tests.