fix test-framework doc

heliaxdev · Nov 9, 2023 · dce7242 · dce7242
1 parent 357b110
commit dce7242
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tools/test-framework/src/docs/walkthroughs/memo.rs b/tools/test-framework/src/docs/walkthroughs/memo.rs
@@ -10,7 +10,7 @@
 //!
 //! ```no_run
 //! # use serde_json as json;
-//! # use ibc_relayer::config::{types::Memo, Config};
+//! # use ibc_relayer::config::{types::Memo, Config, ChainConfig};
 //! # use ibc_test_framework::ibc::denom::derive_ibc_denom;
 //! # use ibc_test_framework::prelude::*;
 //! # use ibc_test_framework::util::random::{random_string, random_u128_range};
@@ -29,7 +29,11 @@
 //! impl TestOverrides for MemoTest {
 //!     fn modify_relayer_config(&self, config: &mut Config) {
 //!         for mut chain in config.chains.iter_mut() {
-//!             chain.memo_prefix = self.memo.clone();
+//!             match chain {
+//!                 ChainConfig::CosmosSdk(chain_config) | ChainConfig::Namada(chain_config) => {
+//!                     chain_config.memo_prefix = self.memo.clone();
+//!                 }
+//!             }
 //!         }
 //!     }
 //! }