Fix for compilation error