diff --git a/koboldai_settings.py b/koboldai_settings.py
index ea6f5823..95f42503 100644
--- a/koboldai_settings.py
+++ b/koboldai_settings.py
@@ -159,7 +159,7 @@ class koboldai_vars(object):
         else:
             method = 1
         context = []
-        token_budget = self.max_length
+        token_budget = self.max_length - self.genamt
         used_world_info = []
         if self.tokenizer is None:
             used_tokens = 99999999999999999999999