bug fix

count -> int(count)