Fixing compile error from last commit