Fixing count=0 bug?