Fix CI build failure