better?

statamic · Dec 2, 2024 · 32e8b02 · 32e8b02
1 parent 8ae4bf6
commit 32e8b02
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/Entries/Entry.php b/src/Entries/Entry.php
@@ -142,9 +142,16 @@ public function collection($collection = null)
         }
 
         $this->computedCallbackCache = null;
-        $this->collection = $collection instanceof \Statamic\Contracts\Entries\Collection ? $collection->handle() : $collection;
 
-        if (is_null($customClass = Collection::findByHandle($this->collection)->class())) {
+        if ($collection instanceof \Statamic\Contracts\Entries\Collection) {
+            $this->collection = $collection->handle();
+            $customClass = $collection->class();
+        } else {
+            $this->collection = $collection;
+            $customClass = Collection::findByHandle($this->collection)->class();
+        }
+
+        if (is_null($customClass)) {
             return $this;
         }