Add count type to Agent check

cool, thanks.