diff --git a/src/dataset_test.py b/src/dataset_test.py
index 3256708..686d963 100644
--- a/src/dataset_test.py
+++ b/src/dataset_test.py
@@ -29,6 +29,6 @@ def test_dataset(self):
     def test_dataloader(self):
         self.assertEqual(len(self.dataloader_test), 2)
         ret = 0
-        for _ in self.dataloader:
+        for _ in self.dataloader_test:
             ret += 1
         self.assertEqual(ret, 2 // 2)