Improve

paritytech · Dec 22, 2024 · 2e6dfbe · 2e6dfbe
1 parent 088c645
commit 2e6dfbe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/substrate/frame/balances/src/lib.rs b/substrate/frame/balances/src/lib.rs
@@ -154,6 +154,8 @@ mod impl_fungible;
 mod tests;
 mod types;
 
+#[cfg(feature = "try-runtime")]
+use frame::try_runtime::prelude::*;
 pub use impl_currency::{NegativeImbalance, PositiveImbalance};
 pub use pallet::*;
 pub use types::{
@@ -547,7 +549,7 @@ pub mod pallet {
 		}
 
 		#[cfg(feature = "try-runtime")]
-		fn try_state(_n: BlockNumberFor<T>) -> Result<(), sp_runtime::TryRuntimeError> {
+		fn try_state(_n: BlockNumberFor<T>) -> Result<(), TryRuntimeError> {
 			Holds::<T, I>::iter_keys().try_for_each(|k| {
 				if Holds::<T, I>::decode_len(k).unwrap_or(0) >
 					T::RuntimeHoldReason::VARIANT_COUNT as usize

diff --git a/substrate/frame/src/lib.rs b/substrate/frame/src/lib.rs
@@ -258,7 +258,9 @@ pub mod prelude {
 
 #[cfg(any(feature = "try-runtime", test))]
 pub mod try_runtime {
-	pub use sp_runtime::TryRuntimeError;
+	pub mod prelude {
+		pub use sp_runtime::TryRuntimeError;
+	}
 }
 
 /// Prelude to be included in the `benchmarking.rs` of a pallet.