Count fix

Count method fix