diff --git a/helios/dataset/dataloader_mp4_dist.py b/helios/dataset/dataloader_mp4_dist.py
index 8fc7529..7aca1d5 100644
--- a/helios/dataset/dataloader_mp4_dist.py
+++ b/helios/dataset/dataloader_mp4_dist.py
@@ -32,34 +32,8 @@
 }
 
 length_bucket_options = {
-    1: [
-        501,
-        481,
-        461,
-        441,
-        421,
-        401,
-        381,
-        361,
-        341,
-        321,
-        301,
-        281,
-        261,
-        241,
-        221,
-        193,
-        181,
-        161,
-        141,
-        121,
-        101,
-        81,
-        61,
-        41,
-        21,
-    ],
-    2: [193, 177, 161, 156, 145, 133, 129, 121, 113, 109, 97, 85, 81, 73, 65, 61, 49, 37, 25],
+    1: [495, 462, 429, 396, 363, 330, 297, 264, 231, 198, 165, 132, 99, 66, 33],
+    2: [165, 132, 99, 66, 33],
 }