potential fix for total_count