diff --git a/train.py b/train.py
index ef45c5dd..ff02e154 100644
--- a/train.py
+++ b/train.py
@@ -210,6 +210,7 @@ def train():
     # Test Dataloader
     if not opt.prebias:
         testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, img_size, batch_size, hyp=hyp,
+                                                                     rect=True,
                                                                      cache_labels=True,
                                                                      cache_images=opt.cache_images),
                                                  batch_size=batch_size,