chore(test): fix failing tests