diff --git a/src/levanter/checkpoint.py b/src/levanter/checkpoint.py
index d6f062ffc..651b84c68 100644
--- a/src/levanter/checkpoint.py
+++ b/src/levanter/checkpoint.py
@@ -27,7 +27,6 @@
 
 from levanter.tensorstore_serialization import tree_deserialize_leaves_tensorstore, tree_serialize_leaves_tensorstore
 from levanter.types import FilterSpec
-# from levanter.trainer import StepInfo
 
 
 logger = logging.getLogger(__name__)