diff --git a/clu/metric_writers/tf/summary_writer.py b/clu/metric_writers/tf/summary_writer.py
index 7e00570..c767d33 100644
--- a/clu/metric_writers/tf/summary_writer.py
+++ b/clu/metric_writers/tf/summary_writer.py
@@ -26,12 +26,12 @@
 from clu.internal import utils
 from clu.metric_writers import interface
 from etils import epy
-import tensorflow as tf
 
 with epy.lazy_imports():
   # pylint: disable=g-import-not-at-top
   from tensorboard.plugins.hparams import api as hparams_api
   from tensorboard.plugins.mesh import summary as mesh_summary  # pylint: disable=line-too-long
+  import tensorflow as tf
   # pylint: enable=g-import-not-at-top