Add count type to Agent check

See reply above