diff --git a/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageSchemaTest.php b/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageSchemaTest.php
index 4055765..2fdf9a1 100644
--- a/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageSchemaTest.php
+++ b/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageSchemaTest.php
@@ -1180,10 +1180,8 @@ public function testRequiresEntityStorageSchemaChanges() {
       ->will($this->returnValue($table_mapping));
 
     $updated_entity_type_definition = $this->getMockBuilder('\Drupal\Core\Entity\ContentEntityTypeInterface')
-      ->disableOriginalConstructor()
       ->getMock();
     $original_entity_type_definition = $this->getMockBuilder('\Drupal\Core\Entity\ContentEntityTypeInterface')
-      ->disableOriginalConstructor()
       ->getMock();
 
     $updated_entity_type_definition->expects($this->any())
@@ -1199,15 +1197,11 @@ public function testRequiresEntityStorageSchemaChanges() {
       ->method('getKey')
       ->willReturn('id');
 
-    // Case 1: Class names change, should not result in required schema change.
+    // Case 1: Storage class changes should not impact this at all.
     $updated_entity_type_definition->expects($this->never())
-      ->method('getStorageClass')
-      // A class that exists, *any* class.
-      ->willReturn('Drupal\Core\Entity\Sql\SqlContentEntityStorage');
+      ->method('getStorageClass');
     $original_entity_type_definition->expects($this->never())
-      ->method('getStorageClass')
-      // A class that exists, *any* class.
-      ->willReturn('\Drupal\Core\Entity\Sql\SqlContentEntityStorageSchema');
+      ->method('getStorageClass');
     $expected = [
       'entity_test' => [
         'primary key' => ['id'],