diff --git a/olive.go b/olive.go
index 298f5e8..bb50f15 100644
--- a/olive.go
+++ b/olive.go
@@ -90,12 +90,13 @@ func (c *Command) AddSubcommand(name, desc string, helpEnabled bool) *Command {
 }
 
 // AddPrimaryArg adds a primary argument to the command
-func (c *Command) AddPrimaryArg(name, desc string, required bool) {
+func (c *Command) AddPrimaryArg(name, desc string, required bool) *PrimaryArgument {
 	if len(c.subcommands) > 0 {
 		log.Fatalf("command `%s` cannot both take a primary argument and have subcommands", c.Name)
 	}
 
 	c.primaryArg = &PrimaryArgument{name: name, desc: desc, required: required}
+	return c.primaryArg
 }
 
 // AddFlag adds a flag to the command