diff --git a/triton_patch/custom_cache_manager.py b/triton_patch/custom_cache_manager.py
index e065bafb7..a923d9e63 100644
--- a/triton_patch/custom_cache_manager.py
+++ b/triton_patch/custom_cache_manager.py
@@ -1,10 +1,7 @@
 import os
-from triton.runtime.cache import (
-    FileCacheManager,
-    default_dump_dir,
-    default_override_dir,
-    default_cache_dir
-)
+
+from triton.runtime.cache import (FileCacheManager, default_cache_dir,
+                                  default_dump_dir, default_override_dir)
 
 
 class CustomCacheManager(FileCacheManager):