fix docs and mocks