fix(test): fix unit test