Fix count function oversight