fix docs and tests