diff --git a/docs/modules/models/llms/examples/llm_caching.ipynb b/docs/modules/models/llms/examples/llm_caching.ipynb
index cec16090..4d591c2c 100644
--- a/docs/modules/models/llms/examples/llm_caching.ipynb
+++ b/docs/modules/models/llms/examples/llm_caching.ipynb
@@ -415,7 +415,7 @@
     "\n",
     "# Avoid multiple caches using the same file, causing different llm model caches to affect each other\n",
     "\n",
-    "def init_gptcache(cache_obj: Cache, llm str):\n",
+    "def init_gptcache(cache_obj: Cache, llm: str):\n",
     "    cache_obj.init(\n",
     "        pre_embedding_func=get_prompt,\n",
     "        data_manager=manager_factory(manager=\"map\", data_dir=f\"map_cache_{llm}\"),\n",