Use total_count if given