diff --git a/langchain/prompts/example_selector/length_based.py b/langchain/prompts/example_selector/length_based.py
index 086b72ce55e..a6f6cd8d235 100644
--- a/langchain/prompts/example_selector/length_based.py
+++ b/langchain/prompts/example_selector/length_based.py
@@ -51,7 +51,7 @@ class LengthBasedExampleSelector(BaseExampleSelector, BaseModel):
         examples = []
         while remaining_length > 0 and i < len(self.examples):
             new_length = remaining_length - self.example_text_lengths[i]
-            if i < 0:
+            if new_length < 0:
                 break
             else:
                 examples.append(self.examples[i])
diff --git a/tests/unit_tests/prompts/test_length_based_example_selector.py b/tests/unit_tests/prompts/test_length_based_example_selector.py
index 45895871b80..38fd689c4e8 100644
--- a/tests/unit_tests/prompts/test_length_based_example_selector.py
+++ b/tests/unit_tests/prompts/test_length_based_example_selector.py
@@ -17,7 +17,7 @@ def selector() -> LengthBasedExampleSelector:
     selector = LengthBasedExampleSelector(
         examples=EXAMPLES,
         example_prompt=prompts,
-        max_length=25,
+        max_length=30,
     )
     return selector