diff --git a/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php b/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php
index 178ac7c5ce..6e681770ad 100644
--- a/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php
+++ b/mailpoet/lib/Automation/Engine/Builder/CreateWorkflowFromTemplateController.php
@@ -3,6 +3,7 @@
 namespace MailPoet\Automation\Engine\Builder;
 
 use MailPoet\Automation\Engine\Data\Workflow;
+use MailPoet\Automation\Engine\Exceptions\InvalidStateException;
 use MailPoet\Automation\Engine\Storage\WorkflowStorage;
 use MailPoet\Automation\Engine\Storage\WorkflowTemplateStorage;
 use MailPoet\UnexpectedValueException;
@@ -32,7 +33,7 @@ class CreateWorkflowFromTemplateController {
     $workflowId = $this->storage->createWorkflow($template->getWorkflow());
     $workflow = $this->storage->getWorkflow($workflowId);
     if (!$workflow) {
-      throw UnexpectedValueException::create()->withMessage('Workflow not found.');
+      throw new InvalidStateException('Workflow not found.');
     }
     return $workflow;
   }