diff --git a/src/_mix b/src/_mix
index c92432c..989adef 100644
--- a/src/_mix
+++ b/src/_mix
@@ -227,11 +227,7 @@ case $state in
          _arguments ':feature:__task_list'
          return
       ;;
-      (test)
-         _arguments ':PATH:_files'
-         return
-      ;;
-      (format)
+      (test|format|run)
          _arguments ':PATH:_files'
          return
       ;;