diff --git a/cps.module b/cps.module
index edc0793..14d4412 100644
--- a/cps.module
+++ b/cps.module
@@ -1167,7 +1167,7 @@ function cps_publish_changeset($changeset, $options = array()) {
       cps_notify_progress($changeset, $total, $processed);
     }
 
-    $processed = cps_publish_changeset_entities($changeset, $total, $processed);
+    $processed = cps_publish_changeset_entities($changeset, $total, $processed, $options);
 
     foreach (cps_get_supported() as $entity_type) {
       $start = 0;
@@ -1238,7 +1238,7 @@ function cps_revert_changeset($changeset_from, $options = array()) {
     if ($options['notify_progress']) {
       cps_notify_progress($changeset_from, $total, $processed);
     }
-    $processed = cps_unpublish_changeset_entities($changeset_from, $changeset_to, $total, $processed);
+    $processed = cps_unpublish_changeset_entities($changeset_from, $changeset_to, $total, $processed, $options);
     if ($options['notify_progress']) {
       cps_notify_progress($changeset_from, $total, $total);
     }
@@ -1284,7 +1284,6 @@ function cps_revert_changeset($changeset_from, $options = array()) {
  *    The updated number of processed entities.
  */
 function cps_unpublish_changeset_entities($changeset_from, $changeset_to, $total = 0, $processed = 0, $options = array()) {
-
   // Merge in defaults.
   $options += array(
     'notify_progress' => TRUE,
diff --git a/includes/forms.inc b/includes/forms.inc
index 771c761..f4f1541 100644
--- a/includes/forms.inc
+++ b/includes/forms.inc
@@ -85,7 +85,7 @@ function cps_changeset_edit_form($form, &$form_state) {
     '#title' => t('Description'),
     '#default_value' => $entity->description,
   );
-  
+
   $form['info']['lock_in_select'] = array(
     '#type' => 'checkbox',
     '#title' => t('Lock in select'),