diff --git a/README.md b/README.md
index 7589c17..99e6906 100755
--- a/README.md
+++ b/README.md
@@ -62,7 +62,13 @@ git clone <repository-url>
 
 You're ready to start developing your chat function. Head over to the [Development](#development) section to learn more.
 
-#### 4. Update the README
+#### 4. Deploy the chat function
+
+You will have to add your API key and LLM model name into the Github repo settings. Under `Secrets and variables/Actions`: the API key must be added as a secret and the LLM model must be added as a variable.
+
+You must ensure the same namings as in your `.env` file. So, make sure to update the `.github/{dev and main}.yml` files with the correct parameter names. 
+
+#### 5. Update the README
 
 In the `README.md` file, change the title and description so it fits the purpose of your chat function.
 
diff --git a/src/module.py b/src/module.py
index 2b7b2b4..02c6f66 100755
--- a/src/module.py
+++ b/src/module.py
@@ -1,6 +1,7 @@
 import time
 from typing import Any
-from lf_toolkit.chat import ChatResult as Result, ChatParams as Params
+from lf_toolkit.chat.result import ChatResult as Result
+from lf_toolkit.chat.params import ChatParams as Params
 
 try:
     from .agents.utils.parse_json_context_to_prompt import parse_json_to_prompt