Fix compile errros