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();
     }