diff --git a/face_swapper/src/data_loader.py b/face_swapper/src/data_loader.py
index 81f73a3..50f3f27 100644
--- a/face_swapper/src/data_loader.py
+++ b/face_swapper/src/data_loader.py
@@ -19,7 +19,7 @@ class DataLoaderVGG(TensorDataset):
 		self.transforms = self.compose_transforms()
 
 	def __getitem__(self, index : int) -> Batch:
-		source_image_path = self.image_paths.get(index)
+		source_image_path = self.image_paths[index]
 
 		if random.random() > self.same_person_probability:
 			return self.prepare_same_person(source_image_path)