fix brocken unit tests