diff --git a/lollms/config.py b/lollms/config.py
index cf4812d..d2725f0 100644
--- a/lollms/config.py
+++ b/lollms/config.py
@@ -532,8 +532,8 @@ class TypedConfig:
                     entry_value = int(entry_value)
                 elif entry_type == "float":
                     entry_value = float(entry_value)
-                elif entry_type == "str" or entry_type == "string":
-                    entry_value = str(entry_value)
+                elif entry_type == "str" or entry_type == "string" or entry_type == "btn":
+                    entry_value = str(entry_value)                   
                 elif entry_type == "bool":
                     entry_value = bool(entry_value)
                 elif entry_type == "list":
diff --git a/lollms/personality.py b/lollms/personality.py
index 4d81517..d3ad0d2 100644
--- a/lollms/personality.py
+++ b/lollms/personality.py
@@ -1330,7 +1330,7 @@ class APScript(StateMachine):
 
 
     def generate(self, prompt, max_size, temperature = None, top_k = None, top_p=None, repeat_penalty=None, callback=None ):
-        self.personality.generate(prompt, max_size, temperature, top_k, top_p, repeat_penalty, callback)
+        return self.personality.generate(prompt, max_size, temperature, top_k, top_p, repeat_penalty, callback)
 
     def run_workflow(self, prompt:str, previous_discussion_text:str="", callback: Callable[[str, MSG_TYPE, dict, list], bool]=None):
         """