Fix compile failure