diff --git a/actions/rules.action.inc b/actions/rules.action.inc
index 31d9dfc..5da3205 100644
--- a/actions/rules.action.inc
+++ b/actions/rules.action.inc
@@ -23,8 +23,12 @@ function views_bulk_operations_rules_action($entity, $context) {
 function views_bulk_operations_rules_action_form($form, &$form_state, $context) {
   require_once DRUPAL_ROOT . '/' . drupal_get_path('module', 'rules') . "/ui/ui.forms.inc";
 
-  $element = rules_action('component_' . $context['action']['key']);
-  $element->form($form, $form_state);
+  $entity_type = $context['action']['type'];
+  $info = entity_get_info($entity_type);
+  $set = rules_action_set(array($entity_type => array('type' => $entity_type, 'label' => $info['label'])));
+  $action = rules_action('component_' . $context['action']['key'], array('VAR:select' => 'entity'));
+  $set->action($action);
+  $action->form($form, $form_state);
 
   // Remove the form element for the "entity" param. It will be passed in manually.
   $entity_key = $context['action']['callback arguments']['entity_key'];