fix bug about count