test: improve test coverage