fixed failing unittest