Fix the compile error