fix matching

opentensor · Nov 6, 2024 · 10dfecb · 10dfecb
1 parent a78f33c
commit 10dfecb
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/runtime/src/lib.rs b/runtime/src/lib.rs
@@ -643,7 +643,7 @@ pub enum ProxyType {
     Transfer,
     SmallTransfer,
     RootWeights,
-    SetCode,
+    SudoUncheckedSetCode,
 }
 // Transfers below SMALL_TRANSFER_LIMIT are considered small transfers
 pub const SMALL_TRANSFER_LIMIT: Balance = 500_000_000; // 0.5 TAO
@@ -716,9 +716,17 @@ impl InstanceFilter<RuntimeCall> for ProxyType {
                 c,
                 RuntimeCall::SubtensorModule(pallet_subtensor::Call::set_root_weights { .. })
             ),
-            ProxyType::SetCode => {
-                matches!(c, RuntimeCall::System(frame_system::Call::set_code { .. }))
-            }
+            ProxyType::SudoUncheckedSetCode => match c {
+                RuntimeCall::Sudo(pallet_sudo::Call::sudo_unchecked_weight { call, weight: _ }) => {
+                    let inner_call: RuntimeCall = *call.clone();
+
+                    matches!(
+                        inner_call,
+                        RuntimeCall::System(frame_system::Call::set_code { .. })
+                    )
+                }
+                _ => false,
+            },
         }
     }
     fn is_superset(&self, o: &Self) -> bool {