diff --git a/prompting/validators/reward/diversity.py b/prompting/validators/reward/diversity.py
index d0a5b77..66314c5 100644
--- a/prompting/validators/reward/diversity.py
+++ b/prompting/validators/reward/diversity.py
@@ -68,7 +68,7 @@ def __init__(self, device: str):
         self.history_reward_bottom_k = 2
         self.historic_embeddings = torch.tensor([]).to(self.device)
         self.history_range = (500, 15500)
-        self.boundary = 0.5
+        self.boundary = 0.2
 
     def get_embeddings(self, sentences: List[str]) -> "torch.FloatTensor":
         """Runs a forward pass through the model.