diff --git a/grain/_src/python/dataset/stats.py b/grain/_src/python/dataset/stats.py
index d350399c4..03b45fb03 100644
--- a/grain/_src/python/dataset/stats.py
+++ b/grain/_src/python/dataset/stats.py
@@ -22,6 +22,7 @@
 import dataclasses
 import enum
 import functools
+import html
 from multiprocessing import queues
 import pprint
 import queue