fix issue with count