Fix compilation error in previous commit