fix brocken tests