Fix possible build error