Fix cargo build error