Small header fix for correct compilation.