diff --git a/common/src/main/java/xyz/jpenilla/tabtps/common/command/Commands.java b/common/src/main/java/xyz/jpenilla/tabtps/common/command/Commands.java
index 4cd4b211..5a274c06 100644
--- a/common/src/main/java/xyz/jpenilla/tabtps/common/command/Commands.java
+++ b/common/src/main/java/xyz/jpenilla/tabtps/common/command/Commands.java
@@ -23,7 +23,7 @@
  */
 package xyz.jpenilla.tabtps.common.command;
 
-import java.util.function.UnaryOperator;
+import java.util.function.Function;
 import org.checkerframework.checker.nullness.qual.NonNull;
 import org.incendo.cloud.Command;
 import org.incendo.cloud.CommandManager;
@@ -43,7 +43,7 @@ public Commands(final @NonNull TabTPS tabTPS, final @NonNull CommandManager<Comm
     return this.commandManager;
   }
 
-  public void registerSubcommand(final @NonNull UnaryOperator<Command.Builder<Commander>> modifier) {
+  public void registerSubcommand(final @NonNull Function<Command.Builder<Commander>, Command.Builder<? extends Commander>> modifier) {
     this.commandManager.command(modifier.apply(this.rootBuilder()));
   }