diff --git a/workflow_extensions.module b/workflow_extensions.module
index 4935dd4..4fa22c7 100755
--- a/workflow_extensions.module
+++ b/workflow_extensions.module
@@ -363,7 +363,10 @@ function _workflow_extensions_replace_with_buttons(&$form, $workflow_name) {
     }
   }
   // Get rid of workflow radio buttons that live inside the fieldset
-  unset($form['workflow'][$workflow_name]);
+//  unset($form['workflow'][$workflow_name]);
+  $form['workflow'][$workflow_name]['#type'] = 'markup';
+  $form['workflow'][$workflow_name]['#markup'] = '<h3>' . $form['workflow'][$workflow_name]['#title'] . '</h3>';
+  
   // If after this the fieldset is empty, remove it altogher
   if (!isset($form['workflow']['workflow_scheduled']) &&
     (!isset($form['workflow']['workflow_comment']) || $form['workflow']['workflow_comment']['#type'] == 'hidden')) {