diff --git a/load_embedding_model.ipynb b/load_embedding_model.ipynb
index 6120b52..95f72e1 100644
--- a/load_embedding_model.ipynb
+++ b/load_embedding_model.ipynb
@@ -166,7 +166,7 @@
       "source": [
         "# Set the model name from Hugging Face and task type\n",
         "hf_model_id='sentence-transformers/all-distilroberta-v1'\n",
-        "tm = TransformerModel(hf_model_id, \"text_embedding\")\n",
+        "tm = TransformerModel(model_id=hf_model_id, task_type="text_embedding")\n",
         "\n",
         "#set the modelID as it is named in Elasticsearch\n",
         "es_model_id = tm.elasticsearch_model_id()\n",