diff --git a/src/wallet/mod.rs b/src/wallet/mod.rs
index 60d1ccef..5b51d43b 100644
--- a/src/wallet/mod.rs
+++ b/src/wallet/mod.rs
@@ -729,7 +729,9 @@ impl Wallet {
     ///
     /// Returns whether the given index was present and then removed from the unused set.
     pub fn mark_used(&mut self, keychain: KeychainKind, index: u32) -> bool {
-        self.tx_graph.index.mark_used(keychain, index)
+        self.tx_graph
+            .index
+            .mark_used(self.map_keychain(keychain), index)
     }
 
     /// Undoes the effect of [`mark_used`] and returns whether the `index` was inserted
@@ -741,7 +743,9 @@ impl Wallet {
     ///
     /// [`mark_used`]: Self::mark_used
     pub fn unmark_used(&mut self, keychain: KeychainKind, index: u32) -> bool {
-        self.tx_graph.index.unmark_used(keychain, index)
+        self.tx_graph
+            .index
+            .unmark_used(self.map_keychain(keychain), index)
     }
 
     /// List addresses that are revealed but unused.