Fixed compile issue for new code