Fixed compile error due to wrong include