chore: add test coverage