Updated a little count bug