Fix for compilation failure