Fix compilation failure of previous commit