--- a/pallets/nft/src/lib.rs +++ b/pallets/nft/src/lib.rs @@ -323,6 +323,8 @@ CollectionNotFound, /// Item not exists. TokenNotFound, + /// Admin not found + AdminNotFound, /// Arithmetic calculation overflow. NumOverflow, /// Account already has admin role. @@ -812,13 +814,11 @@ let sender = ensure_signed(origin)?; Self::check_owner_or_admin_permissions(collection_id, sender)?; + ensure!(>::contains_key(collection_id), Error::::AdminNotFound); - if >::contains_key(collection_id) - { - let mut admin_arr = >::get(collection_id); - admin_arr.retain(|i| *i != account_id); - >::insert(collection_id, admin_arr); - } + let mut admin_arr = >::get(collection_id); + admin_arr.retain(|i| *i != account_id); + >::insert(collection_id, admin_arr); Ok(()) }