added missing includes to count