diff --git a/AttMIL_Training.py b/AttMIL_Training.py
index 9d618ad..2c19fe5 100644
--- a/AttMIL_Training.py
+++ b/AttMIL_Training.py
@@ -171,7 +171,7 @@ def AttMIL_Training(args):
                     
                     model.load_state_dict(torch.load(os.path.join(args.projectFolder, 'RESULTS', 'bestModelFold' + str(foldcounter))))
                     model = model.to(device)
-                    test_dl = dls.test_dl(test_Data)
+                    test_dl = dls.test_dl(test_data)
                     probsList  = Validate_model_AttMIL(model = model, dataloaders = test_dl)
 
                     probs = {}
@@ -269,4 +269,4 @@ def AttMIL_Training(args):
         
         
         
-        
\ No newline at end of file
+