fix(test): Correct unit test bugs