Split into separate trait

src/Http/Livewire/LivewireCollection.php

@@ -12,8 +12,8 @@ class LivewireCollection extends Component
 {
     use Hookable,
         Traits\GenerateParams,
-        Traits\HandleEntriesCount,
         Traits\HandleParams,
+        Traits\HandleTotalEntriesCount,
         Traits\WithPagination;
 
     public $params;

src/Http/Livewire/Traits/HandleTotalEntriesCount.php

@@ -0,0 +1,19 @@
+<?php
+
+namespace Reach\StatamicLivewireFilters\Http\Livewire\Traits;
+
+use Statamic\Entries\EntryCollection;
+
+trait HandleTotalEntriesCount
+{
+    protected function countAllEntries(array $entries): int
+    {
+        if (isset($entries['pagination_total'])) {
+            return (int) $entries['pagination_total'];
+        } elseif ($entries['entries'] instanceof EntryCollection) {
+            return $entries['entries']->count();
+        }
+
+        return 0;
+    }
+}