Fix the build problem