diff --git a/core/modules/media_library/tests/src/Unit/MediaLibrarySelectFormTest.php b/core/modules/media_library/tests/src/Unit/MediaLibrarySelectFormTest.php
index bb11340623..45ee094482 100644
--- a/core/modules/media_library/tests/src/Unit/MediaLibrarySelectFormTest.php
+++ b/core/modules/media_library/tests/src/Unit/MediaLibrarySelectFormTest.php
@@ -56,6 +56,12 @@ public function testViewsForm() {
       ->method('initStyle')
       ->willReturn(TRUE);
 
+    $display_manager = $this->getMockBuilder('\Drupal\views\Plugin\ViewsPluginManager')
+      ->disableOriginalConstructor()
+      ->getMock();
+    $container->set('plugin.manager.views.display', $display_manager);
+    \Drupal::setContainer($container);
+
     $display = $this->createMock(DisplayPluginBase::class);
     $form_state = $this->createMock(FormStateInterface::class);
     $view->result = [$row];