fix(review): fix unit test