Fixing compile bug