badge count fix