diff --git a/core/modules/content_moderation/tests/src/Kernel/ContentModerationStateTest.php b/core/modules/content_moderation/tests/src/Kernel/ContentModerationStateTest.php
index 5fe22ad..00a3379 100644
--- a/core/modules/content_moderation/tests/src/Kernel/ContentModerationStateTest.php
+++ b/core/modules/content_moderation/tests/src/Kernel/ContentModerationStateTest.php
@@ -177,7 +177,7 @@ public function testBasicModeration($entity_type_id) {
     }
 
     // Set an invalid moderation state.
-    $this->setExpectedException(EntityStorageException::class);
+    $this->setExpectedException(\InvalidArgumentException::class);
     $entity->moderation_state->value = 'foobar';
     $entity->save();
   }