Fix compile error from last commit