diff --git a/core/modules/media_entity/tests/src/Functional/MediaUiFunctionalTest.php b/core/modules/media_entity/tests/src/Functional/MediaUiFunctionalTest.php
index 8880398..0fc0121 100644
--- a/core/modules/media_entity/tests/src/Functional/MediaUiFunctionalTest.php
+++ b/core/modules/media_entity/tests/src/Functional/MediaUiFunctionalTest.php
@@ -46,7 +46,7 @@ public function testMediaWithOnlyOneBundle() {
     $assert_session = $this->assertSession();
 
     /** @var \Drupal\media_entity\MediaBundleInterface $bundle */
-    $bundle = $this->drupalCreateMediaBundle();
+    $bundle = $this->drupalCreateMediaBundle(['new_revision' => TRUE]);
 
     $this->drupalGet('media/add');
     $assert_session->statusCodeEquals(200);
@@ -69,6 +69,8 @@ public function testMediaWithOnlyOneBundle() {
     $assert_session->titleEquals($media->label() . ' | Drupal');
 
     // Tests media edit form.
+    $bundle->setNewRevision(FALSE);
+    $bundle->save();
     $media_name2 = $this->randomMachineName();
     $this->drupalGet('media/' . $media_id . '/edit');
     $assert_session->checkboxNotChecked('edit-revision');
@@ -155,9 +157,9 @@ public function testMediaWithMultipleBundles() {
     $assert_session->pageTextContains($first_media_item->label());
 
     // Go to second media item.
-    $this->drupalGet('media/' . $second_media_bundle->id());
+    $this->drupalGet('media/' . $second_media_item->id());
     $assert_session->statusCodeEquals(200);
-    $assert_session->pageTextContains($second_media_bundle->label());
+    $assert_session->pageTextContains($second_media_item->label());
 
   }