updated unittest