fix brocken unit test