From 52d85d7b3dcb8b45f6f85860b1af2bb26f584cd5 Mon Sep 17 00:00:00 2001
From: Rodrigo Primo <rodrigo@automattic.com>
Date: Thu, 15 Sep 2022 19:24:58 -0300
Subject: [PATCH] Remove old Newsletter model from WooCommercePastRevenues

[MAILPOET-4377]
---
 .../DataGenerator/Generators/WooCommercePastRevenues.php      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mailpoet/tests/DataGenerator/Generators/WooCommercePastRevenues.php b/mailpoet/tests/DataGenerator/Generators/WooCommercePastRevenues.php
index 515091448b..3e326ca04e 100644
--- a/mailpoet/tests/DataGenerator/Generators/WooCommercePastRevenues.php
+++ b/mailpoet/tests/DataGenerator/Generators/WooCommercePastRevenues.php
@@ -435,8 +435,8 @@ class WooCommercePastRevenues implements Generator {
       $this->entityManager->refresh($newsletter);
     }
 
-    if ($newsletter->getStatus() === \MailPoet\Models\Newsletter::STATUS_DRAFT) {
-      $newsletter->setStatus(\MailPoet\Models\Newsletter::STATUS_SENT);
+    if ($newsletter->getStatus() === NewsletterEntity::STATUS_DRAFT) {
+      $newsletter->setStatus(NewsletterEntity::STATUS_SENT);
       $newsletter->setSentAt(Carbon::createFromFormat('Y-m-d H:i:s', $sentAt));
       $this->entityManager->flush();
     }