diff --git a/jailbreak_llama.py b/jailbreak_llama.py
index ef4b046..e01e782 100644
--- a/jailbreak_llama.py
+++ b/jailbreak_llama.py
@@ -99,7 +99,7 @@ def main():
     
     with open('benign.json') as f:
         benign_answers = json.load(f)
-    length = len(dataset)
+    length = int(len(dataset)*0.45)
     for i in tqdm(range(length)):
         question = dataset[i]['query']
         adversary = dataset[i]['target']