@@ -212,15 +212,17 @@ impl<T: Config> Pallet<T> {
212
212
// Check the total hotkey stake is the same
213
213
assert_eq ! (
214
214
TotalHotkeyStake :: <T >:: get( taostats_old_hk_acct) ,
215
- old. old_null_stake_taostats + old. old_migration_stake_taostats
215
+ old. old_null_stake_taostats
216
+ . saturating_add( old. old_migration_stake_taostats)
216
217
) ;
217
218
218
219
let new_null_stake_taostats =
219
220
Self :: get_stake_for_coldkey_and_hotkey ( migration_ck_acct, taostats_old_hk_acct) ;
220
221
221
222
assert_eq ! (
222
223
new_null_stake_taostats,
223
- old. old_null_stake_taostats + old. old_migration_stake_taostats
224
+ old. old_null_stake_taostats
225
+ . saturating_add( old. old_migration_stake_taostats)
224
226
) ;
225
227
}
226
228
_ => {
@@ -263,15 +265,17 @@ impl<T: Config> Pallet<T> {
263
265
// Check the total hotkey stake is the same
264
266
assert_eq ! (
265
267
TotalHotkeyStake :: <T >:: get( datura_old_hk_acct) ,
266
- old. old_null_stake_datura + old. old_migration_stake_datura
268
+ old. old_null_stake_datura
269
+ . saturating_add( old. old_migration_stake_datura)
267
270
) ;
268
271
269
272
let new_null_stake_datura =
270
273
Self :: get_stake_for_coldkey_and_hotkey ( migration_ck_acct, datura_old_hk_acct) ;
271
274
272
275
assert_eq ! (
273
276
new_null_stake_datura,
274
- old. old_null_stake_datura + old. old_migration_stake_datura
277
+ old. old_null_stake_datura
278
+ . saturating_add( old. old_migration_stake_datura)
275
279
) ;
276
280
}
277
281
_ => {
0 commit comments