diff --git a/python/tvm/tir/schedule/trace.py b/python/tvm/tir/schedule/trace.py
index d61725e0753a0..ea4410777586a 100644
--- a/python/tvm/tir/schedule/trace.py
+++ b/python/tvm/tir/schedule/trace.py
@@ -285,6 +285,6 @@ def show(self, style: Optional[str] = None, black_format: bool = False) -> None:
 
         if black_format is None:
             env = os.environ.get("TVM_BLACK_FORMAT")
-            black_format = env and int(env)
+            black_format = env and bool(int(env))
 
         cprint(str(self), style=style, black_format=black_format)