diff --git a/tests/streaming/test_dataset.py b/tests/streaming/test_dataset.py
index cf91420f..973ec2e8 100644
--- a/tests/streaming/test_dataset.py
+++ b/tests/streaming/test_dataset.py
@@ -505,15 +505,15 @@ def test_dataset_for_text_tokens_multiple_workers(tmpdir):
     assert len(dataloader) == 10
 
     expected = [
-        [0, 10], 
-        [100, 110], 
-        [20, 30], 
-        [120, 130], 
-        [40, 50], 
-        [140, 150], 
-        [60, 70], 
-        [160, 170], 
-        [80, 90], 
+        [0, 10],
+        [100, 110],
+        [20, 30],
+        [120, 130],
+        [40, 50],
+        [140, 150],
+        [60, 70],
+        [160, 170],
+        [80, 90],
         [180, 190],
     ]