Fix code compilation error