From 184acea42ae46f266427b413ab3ba918fb981619 Mon Sep 17 00:00:00 2001
From: Xavier Lau <xavier@inv.cafe>
Date: Fri, 9 Aug 2024 16:13:54 +0800
Subject: [PATCH] Add addresses

---
 runtime/crab/src/migration.rs     | 6 +++++-
 runtime/darwinia/src/migration.rs | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/runtime/crab/src/migration.rs b/runtime/crab/src/migration.rs
index 60755eafa..25acd0634 100644
--- a/runtime/crab/src/migration.rs
+++ b/runtime/crab/src/migration.rs
@@ -45,7 +45,11 @@ impl frame_support::traits::OnRuntimeUpgrade for CustomOnRuntimeUpgrade {
 }
 
 fn migrate() -> frame_support::weights::Weight {
-	darwinia_staking::migration::migrate_staking_reward_distribution_contract::<Runtime>();
+	if let Some(w) =
+		array_bytes::hex_n_into::<_, _, 20>("0xf1b4f3D438eE2B363C5ba1641A498709ff5780bA")
+	{
+		darwinia_staking::migration::migrate_staking_reward_distribution_contract::<Runtime>(w);
+	}
 
 	// frame_support::weights::Weight::zero()
 	<Runtime as frame_system::Config>::DbWeight::get().reads_writes(10, 10)
diff --git a/runtime/darwinia/src/migration.rs b/runtime/darwinia/src/migration.rs
index 4b4cc1162..6e6d3e45c 100644
--- a/runtime/darwinia/src/migration.rs
+++ b/runtime/darwinia/src/migration.rs
@@ -61,7 +61,11 @@ fn migrate() -> frame_support::weights::Weight {
 	}
 	.remove_storage_values();
 
-	darwinia_staking::migration::migrate_staking_reward_distribution_contract::<Runtime>();
+	if let Some(w) =
+		array_bytes::hex_n_into::<_, _, 20>("0xf1b4f3D438eE2B363C5ba1641A498709ff5780bA")
+	{
+		darwinia_staking::migration::migrate_staking_reward_distribution_contract::<Runtime>(w);
+	}
 
 	// frame_support::weights::Weight::zero()
 	<Runtime as frame_system::Config>::DbWeight::get().reads_writes(10, n + 10)