--- a/rust/hg-pyo3/src/lib.rs	Tue Feb 18 11:45:50 2025 +0100
+++ b/rust/hg-pyo3/src/lib.rs	Tue Feb 18 11:47:51 2025 +0100
@@ -12,6 +12,7 @@
 mod revlog;
 mod store;
 mod transaction;
+mod update;
 mod utils;
 
 #[pymodule]
@@ -29,6 +30,7 @@
     m.add_submodule(&dirstate::init_module(py, &dotted_name)?)?;
     m.add_submodule(&discovery::init_module(py, &dotted_name)?)?;
     m.add_submodule(&revlog::init_module(py, &dotted_name)?)?;
+    m.add_submodule(&update::init_module(py, &dotted_name)?)?;
     m.add("GraphError", py.get_type::<exceptions::GraphError>())?;
     Ok(())
 }