updated unittest according to previous commit