Fixing compile problem