diff --git a/src/Tests/IntegrationTest.php b/src/Tests/IntegrationTest.php
index 8a87fcf..bf80319 100644
--- a/src/Tests/IntegrationTest.php
+++ b/src/Tests/IntegrationTest.php
@@ -171,8 +171,7 @@ class IntegrationTest extends WebTestBase {
     $this->drupalPostForm(NULL, $edit, $this->t('Save and edit'));
 
     $this->assertText($this->t('The index was successfully saved.'));
-    /** @var $index \Drupal\search_api\IndexInterface */
-    $index = Index::load($this->indexId);
+    $index = entity_load('search_api_index', $this->indexId, TRUE);
     $this->assertUrl($index->urlInfo('fields'), array(), 'Correct redirect to index fields page.');
   }