fix: docs and test fixes