diff --git a/tests/debug.py b/tests/debug.py
index 6b27fd33c7..ed8d26cfb4 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -63,7 +63,7 @@ def main():
     assert result == 1, 'amp + ddp model failed to complete'
 
     # test prediction
-    data = model.test_dataloader
+    data = model.val_dataloader
     for batch in data:
         break