diff --git a/storage/aptosdb/src/backup/restore_utils.rs b/storage/aptosdb/src/backup/restore_utils.rs
index 851c3d1489b6b..df8a7400eff62 100644
--- a/storage/aptosdb/src/backup/restore_utils.rs
+++ b/storage/aptosdb/src/backup/restore_utils.rs
@@ -164,8 +164,9 @@ pub(crate) fn save_transactions(
ledger_db.write_schemas(ledger_db_batch)?;
- *state_store.current_state().lock() =
- StateDelta::new_empty_with_version(Some(last_version));
+ state_store
+ .current_state()
+ .set(StateDelta::new_empty_with_version(Some(last_version)));
}
Ok(())
diff --git a/storage/aptosdb/src/db/include/aptosdb_reader.rs b/storage/aptosdb/src/db/include/aptosdb_reader.rs
index f4cd864c909cf..7a29fee5eaf8f 100644
--- a/storage/aptosdb/src/db/include/aptosdb_reader.rs
+++ b/storage/aptosdb/src/db/include/aptosdb_reader.rs
@@ -63,7 +63,7 @@ impl DbReader for AptosDB {
fn get_pre_committed_version(&self) -> Result