Fix the compilation problem