diff --git a/core/modules/views_ui/tests/src/Unit/ViewUIObjectTest.php b/core/modules/views_ui/tests/src/Unit/ViewUIObjectTest.php
index cb26b58..67ec856 100644
--- a/core/modules/views_ui/tests/src/Unit/ViewUIObjectTest.php
+++ b/core/modules/views_ui/tests/src/Unit/ViewUIObjectTest.php
@@ -122,6 +122,12 @@ public function testIsLocked() {
   public function testSerialization() {
     // Set a container so the DependencySerializationTrait has it.
     $container = new ContainerBuilder();
+    $entity_type = $this->getMock('\Drupal\views\ViewEntityInterface');
+    $entity_manager = $this->getMock('\Drupal\Core\Entity\EntityManagerInterface');
+    $entity_manager->expects($this->any())
+      ->method('getDefinition')
+      ->will($this->returnValue($entity_type));
+    $container->set('entity.manager', $entity_manager);
     \Drupal::setContainer($container);
 
     $storage = new View([], 'view');
diff --git a/core/tests/Drupal/Tests/Core/DependencyInjection/DependencySerializationTest.php b/core/tests/Drupal/Tests/Core/DependencyInjection/DependencySerializationTest.php
index 67893a1..1eadc00 100644
--- a/core/tests/Drupal/Tests/Core/DependencyInjection/DependencySerializationTest.php
+++ b/core/tests/Drupal/Tests/Core/DependencyInjection/DependencySerializationTest.php
@@ -20,8 +20,8 @@
 class DependencySerializationTest extends UnitTestCase {
 
   /**
-   * @covers ::__sleep
-   * @covers ::__wakeup
+   * @covers ::serialize
+   * @covers ::unserialize
    */
   public function testSerialization() {
     // Create a pseudo service and dependency injected object.