fix count thing