Use local completion files and directories according to typed command

1 files changed, 4 insertions, 4 deletions

diff --git a/command.go b/command.go
index 7bc0fc5..fc0e28b 100644
--- a/command.go
+++ b/command.go
@@ -17,11 +17,11 @@ func (c *Command) options(args []string) (options []Option, only bool) {
 
 	// remove the first argument, which is the command name
 	args = args[1:]
-
+	last := last(args)
 	// if prev has something that needs to follow it,
 	// it is the most relevant completion
-	if predicate, ok := c.Flags[last(args)]; ok && predicate != nil {
-		return predicate.predict(), true
+	if predicate, ok := c.Flags[last]; ok && predicate != nil {
+		return predicate.predict(last), true
 	}
 
 	sub, options, only := c.searchSub(args)
@@ -41,7 +41,7 @@ func (c *Command) options(args []string) (options []Option, only bool) {
 	}
 
 	// add additional expected argument of the command
-	options = append(options, c.Args.predict()...)
+	options = append(options, c.Args.predict(last)...)
 
 	return
 }