Fix the build on CI