diff --git a/GAN/pi-GAN/datasets.py b/GAN/pi-GAN/datasets.py
index 0da126d..7a1702f 100644
--- a/GAN/pi-GAN/datasets.py
+++ b/GAN/pi-GAN/datasets.py
@@ -80,7 +80,7 @@ def __init__(self,  dataset_path, img_size, **kwargs):
         super().__init__()
         paths = glob.glob(dataset_path)
         self.data = []
-        for path in paths[0:3]:
+        for path in paths:
             self.data = self.data + glob.glob(path + '/rgb/*.png')
         assert len(self.data) > 0, "Can't find data; make sure you specify the path to your dataset"
         self.transform = transforms.Compose(