fix the compile bug