Fixed compile error on last commit