Fix CI breakage