chore(test): add better coverage