chore(test): fix unit tests