diff --git a/lib/pangea/models/bot_options_model.dart b/lib/pangea/models/bot_options_model.dart
index 5038c58af..7ede3c7ab 100644
--- a/lib/pangea/models/bot_options_model.dart
+++ b/lib/pangea/models/bot_options_model.dart
@@ -43,6 +43,8 @@ class BotOptionsModel {
       data[ModelKey.conversationTopic] = topic;
       data[ModelKey.keywords] = keywords;
       data[ModelKey.safetyModeration] = safetyModeration;
+      data[ModelKey.mode] = mode;
+      data[ModelKey.custom] = custom;
       return data;
     } catch (e, s) {
       debugger(when: kDebugMode);
@@ -66,6 +68,12 @@ class BotOptionsModel {
       case ModelKey.safetyModeration:
         safetyModeration = value;
         break;
+      case ModelKey.mode:
+        mode = value;
+        break;
+      case ModelKey.custom:
+        custom = value;
+        break;
       default:
         throw Exception('Invalid key for bot options - $key');
     }