feat: more test coverage