diff --git a/core/modules/system/lib/Drupal/system/Tests/Upgrade/FilterFormatUpgradePathTest.php b/core/modules/system/lib/Drupal/system/Tests/Upgrade/FilterFormatUpgradePathTest.php
index 2725aa9..4c082ee 100644
--- a/core/modules/system/lib/Drupal/system/Tests/Upgrade/FilterFormatUpgradePathTest.php
+++ b/core/modules/system/lib/Drupal/system/Tests/Upgrade/FilterFormatUpgradePathTest.php
@@ -73,7 +73,7 @@ public function testFilterFormatUpgrade() {
     $this->assertNoFieldChecked('edit-5-use-text-format-format-two', 'Use text format format_two permission for role is set correctly.');
 
     // Check that the format has the missing filter.
-    $two = filter_format_load('format_two');
+    $two = entity_load('filter_format', 'format_two');
     $this->assertTrue($two->filters()->has('missing_filter'));
 
     // Try to use a filter format with a missing filter, this should not throw
@@ -86,7 +86,7 @@ public function testFilterFormatUpgrade() {
     $this->assertRaw(t('The %filter filter is missing, and will be removed once this format is saved.', array('%filter' => 'missing_filter')));
     $this->drupalPost(NULL, array(), t('Save configuration'));
     filter_formats_reset();
-    $two = filter_format_load('format_two');
+    $two = entity_load('filter_format', 'format_two');
     $this->assertFalse($two->filters()->has('missing_filter'));
   }