fixed the compile issue