diff --git a/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java b/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java
index 066451e..3ed4aa9 100644
--- a/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java
+++ b/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java
@@ -9,13 +9,18 @@ import java.util.Map;
 public class CommandContextBuilder<S> {
     private final Map<String, ParsedArgument<?>> arguments = Maps.newHashMap();
     private final Map<CommandNode<S>, String> nodes = Maps.newLinkedHashMap();
-    private final S source;
+    private S source;
     private Command<S> command;
 
     public CommandContextBuilder(S source) {
         this.source = source;
     }
 
+    public CommandContextBuilder<S> withSource(S source) {
+        this.source = source;
+        return this;
+    }
+
     public S getSource() {
         return source;
     }