Fix compilation failure