Add count type to Agent check

yup my bad