Fix compile error with last change