diff --git a/crates/index-scheduler/src/scheduler/test_failure.rs b/crates/index-scheduler/src/scheduler/test_failure.rs
index dba0bb072..712fe01a5 100644
--- a/crates/index-scheduler/src/scheduler/test_failure.rs
+++ b/crates/index-scheduler/src/scheduler/test_failure.rs
@@ -8,7 +8,6 @@ use meilisearch_types::milli::update::IndexDocumentsMethod::*;
 use meilisearch_types::milli::update::Setting;
 use meilisearch_types::tasks::Kind;
 use meilisearch_types::tasks::KindWithContent;
-use roaring::RoaringBitmap;
 
 use crate::insta_snapshot::snapshot_index_scheduler;
 use crate::test_utils::Breakpoint::*;
diff --git a/crates/milli/src/update/upgrade/mod.rs b/crates/milli/src/update/upgrade/mod.rs
index a62740e25..5b7fda303 100644
--- a/crates/milli/src/update/upgrade/mod.rs
+++ b/crates/milli/src/update/upgrade/mod.rs
@@ -60,6 +60,7 @@ pub fn upgrade(
             upgrade_path.len() as u32,
         ));
         regenerate_stats |= upgrade.upgrade(wtxn, index, from, progress.clone())?;
+        index.put_version(wtxn, target)?;
         current_version = target;
     }