Fixing compile issue