test(unit): fix test