Fixed compile error in last commit