fixed failing unittest.