Fixed compile error, added missing line