diff --git a/src/compiler/typescript_to_rust/azle_generate_rearchitecture/src/ic/stable_b_tree_map_init.rs b/src/compiler/typescript_to_rust/azle_generate_rearchitecture/src/ic/stable_b_tree_map_init.rs
index 54d92ca88e..f920b10665 100644
--- a/src/compiler/typescript_to_rust/azle_generate_rearchitecture/src/ic/stable_b_tree_map_init.rs
+++ b/src/compiler/typescript_to_rust/azle_generate_rearchitecture/src/ic/stable_b_tree_map_init.rs
@@ -15,7 +15,7 @@ pub fn generate() -> TokenStream {
                 let mut stable_b_tree_maps = stable_b_tree_maps.borrow_mut();
                 stable_b_tree_maps.insert(
                     memory_id,
-                    StableBTreeMap::init_v2(MEMORY_MANAGER_REF_CELL.with(|m| m.borrow().get(MemoryId::new(memory_id))),)
+                    StableBTreeMap::init(MEMORY_MANAGER_REF_CELL.with(|m| m.borrow().get(MemoryId::new(memory_id))),)
                 );
             });