From 738c35891aab1dcab1c9f7115c1e7c99dcfbd8b7 Mon Sep 17 00:00:00 2001
From: Jan Jakes <jan.jakes@automattic.com>
Date: Wed, 31 Jan 2024 16:04:24 +0100
Subject: [PATCH] Simplify getSinglePayloadByClass, remove wrong check

[MAILPOET-5720]
---
 mailpoet/lib/Automation/Engine/Data/StepRunArgs.php | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/mailpoet/lib/Automation/Engine/Data/StepRunArgs.php b/mailpoet/lib/Automation/Engine/Data/StepRunArgs.php
index a5068deff9..72b2bf802a 100644
--- a/mailpoet/lib/Automation/Engine/Data/StepRunArgs.php
+++ b/mailpoet/lib/Automation/Engine/Data/StepRunArgs.php
@@ -113,14 +113,11 @@ class StepRunArgs {
   public function getSinglePayloadByClass(string $class): Payload {
     $payloads = [];
     foreach ($this->subjectEntries as $entries) {
-      if (count($entries) > 1) {
-        throw Exceptions::multiplePayloadsFound($class, $this->automationRun->getId());
-      }
-
-      $entry = $entries[0];
-      $payload = $entry->getPayload();
-      if (get_class($payload) === $class) {
-        $payloads[] = $payload;
+      foreach ($entries as $entry) {
+        $payload = $entry->getPayload();
+        if (get_class($payload) === $class) {
+          $payloads[] = $payload;
+        }
       }
     }