diff --git a/d2go/runner/default_runner.py b/d2go/runner/default_runner.py
index 80384149..aaacbae7 100644
--- a/d2go/runner/default_runner.py
+++ b/d2go/runner/default_runner.py
@@ -39,7 +39,7 @@
 from d2go.trainer.fsdp import get_grad_scaler
 from d2go.trainer.helper import parse_precision_from_string
 from d2go.utils.flop_calculator import attach_profilers
-from d2go.utils.helper import D2Trainer, TensorboardXWriter
+from d2go.utils.helper import D2Trainer, TensorBoardXWriter
 from d2go.utils.misc import get_tensorboard_log_dir
 from d2go.utils.visualization import DataLoaderVisWrapper, VisualizationEvaluator
 from detectron2.checkpoint import DetectionCheckpointer, PeriodicCheckpointer
@@ -72,7 +72,7 @@
 
 @lru_cache()
 def _get_tbx_writer(log_dir, window_size=20):
-    ret = TensorboardXWriter(log_dir, window_size=window_size)
+    ret = TensorBoardXWriter(log_dir, window_size=window_size)
     ALL_TB_WRITERS.append(ret)
     return ret
 
diff --git a/d2go/utils/helper.py b/d2go/utils/helper.py
index b17f6777..90f44556 100644
--- a/d2go/utils/helper.py
+++ b/d2go/utils/helper.py
@@ -21,7 +21,7 @@
     PascalVOCDetectionEvaluator,
     SemSegEvaluator,
 )
-from detectron2.utils.events import TensorboardXWriter
+from detectron2.utils.events import TensorBoardXWriter
 from mobile_cv.common.misc.oss_utils import fb_overwritable
 
 T = TypeVar("T")
@@ -33,7 +33,7 @@
     "run_once",
     "retryable",
     "get_dir_path",
-    "TensorboardXWriter",  # TODO: move to D2Go's vis utils if needed
+    "TensorBoardXWriter",  # TODO: move to D2Go's vis utils if needed
     "D2Trainer",  # TODO: move to trainer folder
 ]