diff --git a/test.py b/test.py
index 85181a212..70a0c5f67 100644
--- a/test.py
+++ b/test.py
@@ -61,7 +61,7 @@ def test_net(save_folder, net, cuda, testset, transform, thresh):
         pred_num = 0
         for i in range(detections.size(1)):
             j = 0
-            while detections[0, i, j, 0] >= 0.6:
+            while detections[0, i, j, 0] >= thresh:
                 if pred_num == 0:
                     with open(filename, mode='a') as f:
                         f.write('PREDICTIONS: '+'\n')