fix count feature