Fix compile issue