Fix count method error