Fix the compile warning