azalea-brigadier now allows commands to return a Result

1 files changed, 5 insertions, 6 deletions

diff --git a/azalea/examples/testbot/commands/combat.rs b/azalea/examples/testbot/commands/combat.rs
index af147e96..617ac1bd 100644
--- a/azalea/examples/testbot/commands/combat.rs
+++ b/azalea/examples/testbot/commands/combat.rs
@@ -1,22 +1,21 @@
 use azalea::brigadier::prelude::*;
-use parking_lot::Mutex;
 
-use super::{CommandSource, Ctx};
-use crate::State;
+use super::Ctx;
+use crate::{State, commands::Dispatcher};
 
-pub fn register(commands: &mut CommandDispatcher<Mutex<CommandSource>>) {
+pub fn register(commands: &mut Dispatcher) {
     commands.register(
         literal("killaura").then(argument("enabled", bool()).executes(|ctx: &Ctx| {
             let enabled = get_bool(ctx, "enabled").unwrap();
             let source = ctx.source.lock();
             let bot = source.bot.clone();
-            bot.query_self::<&mut State, _>(|mut state| state.killaura = enabled);
+            bot.query_self::<&mut State, _>(|mut state| state.killaura = enabled)?;
             source.reply(if enabled {
                 "Enabled killaura"
             } else {
                 "Disabled killaura"
             });
-            1
+            Ok(1)
         })),
     );
 }