Fix issue where passing args to count would not behave correctly