trying to fix count