diff --git a/core/tests/Drupal/Tests/Core/Config/Entity/EntityDisplayModeBaseUnitTest.php b/core/tests/Drupal/Tests/Core/Config/Entity/EntityDisplayModeBaseUnitTest.php
index 733afc5..4f14454 100644
--- a/core/tests/Drupal/Tests/Core/Config/Entity/EntityDisplayModeBaseUnitTest.php
+++ b/core/tests/Drupal/Tests/Core/Config/Entity/EntityDisplayModeBaseUnitTest.php
@@ -109,7 +109,7 @@ public function testCalculateDependencies() {
    * @covers ::getTargetEntityTypeId()
    */
   public function testGetTargetEntityTypeId() {
-    $display = $this->getMockForAbstractClass('\Drupal\entity\EntityDisplayModeBase', array(array(), $this->entityType));
+    $display = $this->getMockForAbstractClass('Drupal\Core\Entity\EntityDisplayModeBase', array(array(), $this->entityType));
     $display->setTargetEntityTypeId('test');
     $this->assertEquals('test', $display->getTargetEntityTypeId());
   }
@@ -130,7 +130,7 @@ public function testSetLabel() {
       ->method('getKey')
       ->with('label')
       ->will($this->returnValue('label'));
-    $display = $this->getMockForAbstractClass('\Drupal\entity\EntityDisplayModeBase', array(array(), $this->entityType));
+    $display = $this->getMockForAbstractClass('Drupal\Core\Entity\EntityDisplayModeBase', array(array(), $this->entityType));
     $display->setLabel('test');
     $this->assertEquals('test', $display->label());
   }