chore(test): update tests