diff --git a/core/modules/content_moderation/src/Tests/UpgradeFrom2xTest.php b/core/modules/content_moderation/src/Tests/UpgradeFrom3xTest.php
similarity index 79%
rename from core/modules/content_moderation/src/Tests/UpgradeFrom2xTest.php
rename to core/modules/content_moderation/src/Tests/UpgradeFrom3xTest.php
index 45082b9..3be693e 100644
--- a/core/modules/content_moderation/src/Tests/UpgradeFrom2xTest.php
+++ b/core/modules/content_moderation/src/Tests/UpgradeFrom3xTest.php
@@ -2,7 +2,7 @@
 
 namespace Drupal\content_moderation\Tests;
 
-use Drupal\system\Tests\Update\UpdatePathTestBase;
+use Drupal\FunctionalTests\Update\UpdatePathTestBase;
 use Drupal\workflows\Entity\Workflow;
 
 /**
@@ -10,7 +10,7 @@
  *
  * @group content_moderation
  */
-class UpgradeFrom2xTest extends UpdatePathTestBase {
+class UpgradeFrom3xTest extends UpdatePathTestBase {
 
   /**
    * Profile to install.
@@ -41,6 +41,7 @@ public function testUpdate() {
     $type = $workflow->getTypePlugin();
     $published = $type->getState('published');
     $draft = $type->getState('draft');
+    $archived = $type->getState('archived');
     $transitions = $type->getTransitions();
 
     $this->assertTrue($published->isDefaultRevisionState());
@@ -48,6 +49,10 @@ public function testUpdate() {
     $this->assertFalse($draft->isPublishedState());
     $this->assertFalse($draft->isDefaultRevisionState());
     $this->assertEqual(5, count($transitions));
+
+    $this->assertEquals($draft->label(), 'Draft');
+    $this->assertEquals($published->label(), 'Published');
+    $this->assertEquals($archived->label(), 'Archived');
   }
 
 }