quick gross fix to count issue