Fix compiling error pt2