test: fix test coverage