diff --git a/tools/protocol-schema-check/src/main.rs b/tools/protocol-schema-check/src/main.rs
index 9422dfcb76f..23d30b3418b 100644
--- a/tools/protocol-schema-check/src/main.rs
+++ b/tools/protocol-schema-check/src/main.rs
@@ -18,6 +18,7 @@ use near_vm_runner::*;
 
 use near_schema_checker_lib::{FieldName, FieldTypeInfo, ProtocolSchema, ProtocolSchemaInfo};
 use near_stable_hasher::StableHasher;
+use near_store::epoch_info_aggregator::EpochInfoAggregator;
 use std::any::TypeId;
 use std::collections::{BTreeMap, HashSet};
 use std::fs;