fix duration check

paritytech · Sep 13, 2022 · 6cb73be · 6cb73be
1 parent ea0f3e4
commit 6cb73be
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/frame/safe-mode/src/lib.rs b/frame/safe-mode/src/lib.rs
@@ -196,8 +196,9 @@ pub mod pallet {
 		/// Emits an [`Event::Enabled`] event on success.
 		#[pallet::weight(0)]
 		pub fn force_enable(origin: OriginFor<T>) -> DispatchResult {
-			let duration = T::EnableOrigin::ensure_origin(origin)?;
+			T::EnableOrigin::ensure_origin(origin)?;
 
+			let duration = T::EnableDuration::get();
 			Self::do_enable(None, duration)
 		}
 
@@ -219,8 +220,9 @@ pub mod pallet {
 		/// Can only be called by the [`Config::ExtendOrigin`] origin.
 		#[pallet::weight(0)]
 		pub fn force_extend(origin: OriginFor<T>) -> DispatchResult {
-			let duration = T::ExtendOrigin::ensure_origin(origin)?;
+			T::ExtendOrigin::ensure_origin(origin)?;
 
+			let duration = T::ExtendDuration::get();
 			Self::do_extend(None, duration)
 		}