diff --git a/search_api_autocomplete.module b/search_api_autocomplete.module
index 9daa5d7..dee6cf3 100644
--- a/search_api_autocomplete.module
+++ b/search_api_autocomplete.module
@@ -151,8 +151,11 @@ function search_api_autocomplete_form_views_exposed_form_alter(array &$form, For
  * @see \Drupal\search_api_autocomplete\Plugin\search_api_autocomplete\search\Page
  */
 function search_api_autocomplete_form_search_api_page_block_form_alter(array &$form, FormStateInterface $form_state) {
-  $id = 'search_api_page_' . $form_state->getBuildInfo()['args'][0]['search_api_page'];
-  $search = Search::load($id);
+  $page_id = $form_state->getBuildInfo()['args'][0]['search_api_page'];
+  /** @var \Drupal\search_api_autocomplete\Entity\SearchStorage $search_storage */
+  $search_storage = \Drupal::entityTypeManager()
+    ->getStorage('search_api_autocomplete_search');
+  $search = $search_storage->loadBySearchPlugin('page:' . $page_id);
 
   if ($search && $search->status()) {
     \Drupal::getContainer()