diff --git a/tools/inference.py b/tools/inference.py
index 80652705..b4ecf20b 100644
--- a/tools/inference.py
+++ b/tools/inference.py
@@ -317,7 +317,7 @@ def after_test_epoch(self, _, metrics=None):
         # TODO: Support '.pickel' dump
         runner = Infernce(
             args.checkpoint,
-            dataloader=runner.test_dataloader,
+            dataloader=runner.val_dataloader,
             cfg=cfg,
             runner=runner,
             source=args.source,