diff --git a/pallets/parachain-staking/src/migrations.rs b/pallets/parachain-staking/src/migrations.rs
index b5fc29e189..4c689e451d 100644
--- a/pallets/parachain-staking/src/migrations.rs
+++ b/pallets/parachain-staking/src/migrations.rs
@@ -80,6 +80,8 @@ impl<T: Config> OnRuntimeUpgrade for MigrateParachainBondConfig<T> {
 
 	#[cfg(feature = "try-runtime")]
 	fn post_upgrade(state: Vec<u8>) -> Result<(), sp_runtime::DispatchError> {
+		use frame_support::ensure;
+
 		let old_state: OldParachainBondConfig<T::AccountId> =
 			parity_scale_codec::Decode::decode(&mut &state[..])
 				.map_err(|_| sp_runtime::DispatchError::Other("Failed to decode old state"))?;