diff --git a/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py b/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py
index a5f8ae96..7f98a1d6 100644
--- a/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py
+++ b/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py
@@ -14,8 +14,6 @@
 
 p = Parea(api_key=os.getenv("PAREA_API_KEY"))
 
-NUM_INTERACTIONS = 5
-
 
 class Person(BaseModel):
     name: str
@@ -28,6 +26,7 @@ class Email(BaseModel):
 
 
 mock_DB: dict[str, Email] = {}
+NUM_INTERACTIONS = 5
 
 
 def call_llm(messages: List[Message]) -> CompletionResponse: