Fix compile error due to the last commit