diff --git a/src/chatgpt.ts b/src/chatgpt.ts
index 67cb57e..e1b04cf 100644
--- a/src/chatgpt.ts
+++ b/src/chatgpt.ts
@@ -205,7 +205,7 @@ class Chat {
     this.top_p = top_p;
     this.enable_top_p = enable_top_p;
     this.presence_penalty = presence_penalty;
-    this.presence_penalty_enabled = presence_penalty_enabled
+    this.presence_penalty_enabled = presence_penalty_enabled;
     this.frequency_penalty = frequency_penalty;
     this.frequency_penalty_enabled = frequency_penalty_enabled;
     this.json_mode = json_mode;
diff --git a/src/types/newChatstore.ts b/src/types/newChatstore.ts
index db6e7e5..0501152 100644
--- a/src/types/newChatstore.ts
+++ b/src/types/newChatstore.ts
@@ -35,6 +35,7 @@ interface NewChatStoreOptions {
   image_gen_key?: string;
   json_mode?: boolean;
   logprobs?: boolean;
+  maxTokens?: number;
 }
 
 export const newChatStore = (options: NewChatStoreOptions): ChatStore => {
@@ -52,7 +53,9 @@ export const newChatStore = (options: NewChatStoreOptions): ChatStore => {
     maxTokens: getDefaultParams(
       "max",
       models[getDefaultParams("model", options.model ?? DefaultModel)]
-        ?.maxToken ?? 2048
+        ?.maxToken ??
+        options.maxTokens ??
+        2048
     ),
     maxGenTokens: 2048,
     maxGenTokens_enabled: false,