[maybe]

Author: tarsius

lisp/transient.el

@@ -1667,9 +1667,12 @@ See info node `(transient)Modifying Existing Transients'."
      ((seq-some #'listp children)
       (when (stringp loc)
         (setq loc (kbd loc)))
-      (and-let* ((suffix (seq-find (lambda (s) (transient--loc-match-p s loc))
-                                   children)))
-        (list suffix group)))
+      (if-let* ((suffix (seq-find (lambda (s) (transient--loc-match-p s loc))
+                                  children)))
+          (list suffix group)
+        (and (symbolp loc)
+             (memq loc children)
+             (list loc group))))
      ((seq-some (lambda (g) (transient--get-suffix g loc)) children))
      ((symbolp loc)
       (and (memq loc children)
@@ -1680,7 +1683,8 @@ See info node `(transient)Modifying Existing Transients'."
        (let* ((props (transient--suffix-props suffix))
               (cmd (plist-get props :command)))
          (if (symbolp loc)
-             (eq cmd loc)
+             (or (eq cmd loc)
+                 (eq suffix loc))
            (equal (kbd (or (plist-get props :key)
                            (transient--command-key cmd)))
                   (kbd loc))))))