diff --git a/streaming/base/util.py b/streaming/base/util.py
index db3875f65..0dc12dcd9 100644
--- a/streaming/base/util.py
+++ b/streaming/base/util.py
@@ -20,7 +20,7 @@
 
 import torch.distributed as dist
 
-from streaming.base.constant import SHM_TO_CLEAN, BARRIER_FILELOCK, CACHE_FILELOCK
+from streaming.base.constant import BARRIER_FILELOCK, CACHE_FILELOCK, SHM_TO_CLEAN
 from streaming.base.distributed import get_local_rank, maybe_init_dist
 from streaming.base.format.index import get_index_basename
 from streaming.base.shared.prefix import _get_path