diff --git a/oa_worktracker.module b/oa_worktracker.module
index 01c3e01..5ca7b15 100644
--- a/oa_worktracker.module
+++ b/oa_worktracker.module
@@ -146,7 +146,8 @@ function _oa_worktracker_check_section_options(&$options) {
       // TODO: we need to cache this somehow for performance!
       $buttons = array();
       _oa_buttons_get_node_command_buttons(node_load($nid), $buttons);
-      if (!isset($buttons['oa_worktracker_task'])) {
+      $key = current(array_keys($buttons));
+      if (strpos($key, 'oa_worktracker_task') === FALSE) {
         unset($options[$nid]);
       }
     }