Fix CI build failed