Add count type to Agent check

let's remove these notes before merging (sorry for flip-flopping on this one) we can do in the future.