chore(test): fix tests