diff --git a/python/dgl/graphbolt/internal_utils.py b/python/dgl/graphbolt/internal_utils.py
index 1efdced99d0a..53e7ed8ed5f6 100644
--- a/python/dgl/graphbolt/internal_utils.py
+++ b/python/dgl/graphbolt/internal_utils.py
@@ -17,7 +17,7 @@
     from setuptools.extern.packaging import version
 
 
-@functools.cache
+@functools.lru_cache(maxsize=None)
 def is_wsl(v: str = platform.uname().release) -> int:
     """Detects if Python is running in WSL"""