fix(test): fix unit tests