diff --git a/lib/Drupal/entityreference/Tests/entityReferenceSelectionAccessTest.php b/lib/Drupal/entityreference/Tests/entityReferenceSelectionAccessTest.php
index b946929..f8e1e99 100644
--- a/lib/Drupal/entityreference/Tests/entityReferenceSelectionAccessTest.php
+++ b/lib/Drupal/entityreference/Tests/entityReferenceSelectionAccessTest.php
@@ -8,6 +8,9 @@
 namespace Drupal\entityreference\Tests;
 
 use Drupal\simpletest\WebTestBase;
+use Drupal\node\Entity\Node;
+use Drupal\user\Entity\User;
+use Drupal\comment\Entity\Comment;
 
 
 /**
@@ -84,7 +87,7 @@ class entityReferenceSelectionAccessTest extends WebTestBase {
     $nodes = array();
     $node_labels = array();
     foreach ($node_values as $key => $values) {
-      $node = entity_create('node', $values);
+      $node = Node::create($values);
       $node->save();
       $nodes[$key] = $node;
       $node_labels[$key] = check_plain($node->label());
@@ -208,7 +211,7 @@ class entityReferenceSelectionAccessTest extends WebTestBase {
     $user_labels = array();
     foreach ($user_values as $key => $values) {
       if (is_array($values)) {
-        $account = entity_create('user', $values);
+        $account = User::create($values);
         $account->save();
       }
       else {
@@ -325,7 +328,7 @@ class entityReferenceSelectionAccessTest extends WebTestBase {
     );
     $nodes = array();
     foreach ($node_values as $key => $values) {
-      $node = entity_create('node', $values);
+      $node = Node::create($values);
       $node->save();
       $nodes[$key] = $node;
     }
@@ -363,7 +366,7 @@ class entityReferenceSelectionAccessTest extends WebTestBase {
     $comments = array();
     $comment_labels = array();
     foreach ($comment_values as $key => $values) {
-      $comment = entity_create('comment', $values);
+      $comment = Comment::create($values);
       $comment->save();
       $comments[$key] = $comment;
       $comment_labels[$key] = check_plain($comment->label());