Fix compile errror