Fix the compile problem