Fixing the compilation failure