Include fix for compilation error