HP-2496: fixing Psalm

Author: VadymHrechukha

src/product/behavior/BehaviorRegistry.php

@@ -6,8 +6,15 @@
 
 use hiqdev\php\billing\product\trait\HasLockInterface;
 
+/**
+ * @template TParentCollection
+ * @implements HasBehaviorsInterface<TParentCollection>
+ */
 abstract class BehaviorRegistry implements HasLockInterface, HasBehaviorsInterface
 {
+    /**
+     * @return BehaviorCollectionInterface<TParentCollection>
+     */
     abstract protected function getBehaviorCollection(): BehaviorCollectionInterface;
 
     public function hasBehavior(string $behaviorClassName): bool
@@ -23,7 +30,7 @@ public function hasBehavior(string $behaviorClassName): bool
 
     public function findBehaviorByClass(string $class)
     {
-        foreach ($this->withBehaviors() as $behavior) {
+        foreach ($this->getBehaviorCollection() as $behavior) {
             if ($behavior instanceof $class) {
                 return $behavior;
             }