diff --git a/astropylibrarian/algolia/client.py b/astropylibrarian/algolia/client.py
index 7a1f48a..1cc448c 100644
--- a/astropylibrarian/algolia/client.py
+++ b/astropylibrarian/algolia/client.py
@@ -67,10 +67,10 @@ class AlgoliaIndex(BaseAlgoliaIndex):
         Name of the Algolia index.
     """
 
-    async def __aenter__(self) -> SearchClient:
+    async def __aenter__(self) -> "AlgoliaIndex":
         self._logger.debug("Opening algolia client")
         self.algolia_client = SearchClient(self.app_id, self._key)
-        return self.algolia_client
+        return self
 
     async def __aexit__(
         self,