Fixed the compilation error