Fix the compilation error