chore(test): better test coverage