diff --git a/test/test_dataloader.py b/test/test_dataloader.py
index 825689ffced6..a356211747ef 100644
--- a/test/test_dataloader.py
+++ b/test/test_dataloader.py
@@ -107,7 +107,7 @@ def __getitem__(self, key):
             def __len__(self):
                 return len(self.data)
 
-        x = [1, 2, 3, 4, 5]
+        x = [1,2, 3, 4, 5]
         dataset = CustomDataset(self, x)
         dataset = random_split(dataset, [5])[0]
         data_loader = DataLoader(dataset)