Added additional test; changed count to find 

* Added one additional test
* Changed some syntax to use find, this is unrelated to the original PR but since I was in here I changed it. I prefer find, I don't know why I used count originally.
(#3178)