diff --git a/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageTest.php b/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageTest.php
index 3589a63..5b5d5ee 100644
--- a/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageTest.php
+++ b/core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageTest.php
@@ -123,7 +123,7 @@ protected function setUp() {
   public function testGetBaseTable($base_table, $expected) {
     $this->entityType->expects($this->once())
       ->method('getBaseTable')
-      ->will($this->returnValue('entity_test'));
+      ->willReturn($base_table);
 
     $this->setUpEntityStorage();