Fix the compile issue