1 files changed, 6 insertions, 2 deletions
diff --git a/azalea-brigadier/src/builder/argument_builder.rs b/azalea-brigadier/src/builder/argument_builder.rs
index d26b2a8a..38ccc98c 100755
--- a/azalea-brigadier/src/builder/argument_builder.rs
+++ b/azalea-brigadier/src/builder/argument_builder.rs
@@ -1,4 +1,8 @@
-use crate::{context::CommandContext, modifier::RedirectModifier, tree::CommandNode};
+use crate::{
+    context::CommandContext,
+    modifier::RedirectModifier,
+    tree::{Command, CommandNode},
+};
 
 use super::{literal_argument_builder::Literal, required_argument_builder::Argument};
 use std::{cell::RefCell, fmt::Debug, rc::Rc};
@@ -13,7 +17,7 @@ pub enum ArgumentBuilderType {
 pub struct ArgumentBuilder<S> {
     arguments: CommandNode<S>,
 
-    command: Option<Rc<dyn Fn(&CommandContext<S>) -> i32>>,
+    command: Command<S>,
     requirement: Rc<dyn Fn(Rc<S>) -> bool>,
     target: Option<Rc<RefCell<CommandNode<S>>>>,