Fix the build failure