More compilation fix