diff --git a/src/ElementSubmit.php b/src/ElementSubmit.php
index 0abde9b..c3bd9b6 100644
--- a/src/ElementSubmit.php
+++ b/src/ElementSubmit.php
@@ -31,7 +31,7 @@ class ElementSubmit {
     $form['#ief_element_submit_attached'] = TRUE;
 
     // Entity form actions.
-    foreach (['submit', 'publish', 'unpublish'] as $action) {
+    foreach (Element::children($form['actions']) as $action) {
       if (!empty($form['actions'][$action])) {
         self::addCallback($form['actions'][$action], $form);
       }