Fix compile bugs