diff --git a/core/modules/shortcut/src/Tests/ShortcutLinksTest.php b/core/modules/shortcut/src/Tests/ShortcutLinksTest.php
index c5ad339..cf2d732 100644
--- a/core/modules/shortcut/src/Tests/ShortcutLinksTest.php
+++ b/core/modules/shortcut/src/Tests/ShortcutLinksTest.php
@@ -6,6 +6,7 @@
  */
 
 namespace Drupal\shortcut\Tests;
+use Drupal\shortcut\Entity\Shortcut;
 use Drupal\shortcut\Entity\ShortcutSet;
 
 /**
@@ -73,7 +74,7 @@ public function testShortcutLinkAdd() {
       // Test the node routes with parameters.
       if (strpos($entity->route_name->value, 'node.') === 0) {
         $entity->save();
-        $loaded = entity_load('shortcut', $entity->id());
+        $loaded = Shortcut::load($entity->id());
         $this->assertEqual($entity->route_name->value, $loaded->route_name->value);
         $this->assertEqual($entity->get('route_parameters')->first()->getValue(), $loaded->get('route_parameters')->first()->getValue());
       }