diff --git a/pallets/maintenance-mode/src/types.rs b/pallets/maintenance-mode/src/types.rs
index 9ad2bb86..0f20dbb4 100644
--- a/pallets/maintenance-mode/src/types.rs
+++ b/pallets/maintenance-mode/src/types.rs
@@ -22,6 +22,8 @@ use frame_support::{
 	traits::{OffchainWorker, OnFinalize, OnIdle, OnInitialize, OnRuntimeUpgrade},
 	weights::Weight,
 };
+#[cfg(feature = "try-runtime")]
+use sp_runtime::TryRuntimeError;
 use sp_std::marker::PhantomData;
 #[cfg(feature = "try-runtime")]
 use sp_std::vec::Vec;
@@ -93,6 +95,15 @@ where
 		}
 	}
 
+	#[cfg(feature = "try-runtime")]
+	fn try_on_runtime_upgrade(checks: bool) -> Result<Weight, TryRuntimeError> {
+		if Pallet::<T>::maintenance_mode() {
+			T::MaintenanceExecutiveHooks::try_on_runtime_upgrade(checks)
+		} else {
+			T::NormalExecutiveHooks::try_on_runtime_upgrade(checks)
+		}
+	}
+
 	#[cfg(feature = "try-runtime")]
 	fn pre_upgrade() -> Result<Vec<u8>, sp_runtime::DispatchError> {
 		if Pallet::<T>::maintenance_mode() {