Fixed compile error from last commit