diff --git a/og_workflow.module b/og_workflow.module
index 777d6e7..37d9b90 100755
--- a/og_workflow.module
+++ b/og_workflow.module
@@ -582,8 +582,9 @@ function og_workflow_user_roles_for_node($node, $user=NULL) {
     $target_groups = og_get_entity_groups('node', $node);
   }
   if (!empty($target_groups)) {
-    foreach (array_values($target_groups['node']) as $group) {
-      $groups[$group] = array_keys(og_get_user_roles('node', $group, $user->uid));
+    $entity_type = key($target_groups);
+    foreach (array_values($target_groups[$entity_type]) as $group) {
+      $groups[$group] = array_keys(og_get_user_roles($entity_type, $group, $user->uid));
     }
   }
   // Check to make sure we are at node/add or else the call menu_get_item() call