diff --git a/read_cifar.py b/read_cifar.py
index b856ea42df9547ca4fd7f07c8a3540a3167ef5b6..12cd6b5df156481dd9c3b07278e2940be0e121a6 100644
--- a/read_cifar.py
+++ b/read_cifar.py
@@ -24,7 +24,7 @@ def read_cifar (batch_dir) :
     data_batches = []
     label_batches = []
     
-    for i in range(1,2) :
+    for i in range(1, 6) :
         batch_filename = f'data_batch_{i}'
         batch_path = os.path.join(batch_dir, batch_filename)
         data, labels = read_cifar_batch(batch_path)