diff --git a/src/bia_bob/_machinery.py b/src/bia_bob/_machinery.py
index 93dd2a6..454e2de 100644
--- a/src/bia_bob/_machinery.py
+++ b/src/bia_bob/_machinery.py
@@ -207,6 +207,7 @@ def init_assistant(model=None, auto_execute:bool = False, variables:dict=None, e
     config = {
         "model": DEFAULT_MODEL,
         "vision_model": DEFAULT_VISION_MODEL,
+        "endpoint": None,
     }
 
     # load config from disk
@@ -222,10 +223,13 @@ def init_assistant(model=None, auto_execute:bool = False, variables:dict=None, e
         model = config["model"]
     if vision_model is None:
         vision_model = config["vision_model"]
+    if endpoint is None:
+        endpoint = config["endpoint"]
 
     # store config to disk
     config["model"] = model
     config["vision_model"] = vision_model
+    config["endpoint"] = endpoint
     with open(config_filename, 'w') as file:
         yaml.dump(config, file, default_flow_style=False)