--- a/rust/hg-pyo3/src/dirstate.rs	Tue Feb 04 10:42:26 2025 +0100
+++ b/rust/hg-pyo3/src/dirstate.rs	Tue Feb 04 13:36:42 2025 +0100
@@ -20,7 +20,7 @@
     DirstateMapKeysIterator,
 };
 mod copy_map;
-use copy_map::CopyMap;
+use copy_map::{CopyMap, CopyMapItemsIterator, CopyMapKeysIterator};
 
 pub fn init_module<'py>(
     py: Python<'py>,
@@ -35,5 +35,7 @@
     m.add_class::<DirstateMapKeysIterator>()?;
     m.add_class::<DirstateMapItemsIterator>()?;
     m.add_class::<CopyMap>()?;
+    m.add_class::<CopyMapKeysIterator>()?;
+    m.add_class::<CopyMapItemsIterator>()?;
     Ok(m)
 }