refactor: more coverage