fixed the compilation error