diff --git a/workflow_extensions.module b/workflow_extensions.module
index a7fb06e..9e95e84 100755
--- a/workflow_extensions.module
+++ b/workflow_extensions.module
@@ -211,7 +211,7 @@ function _workflow_extensions_replace_with_buttons(&$form, $workflow_name) {
     if ($sid != $current_sid) {
       // Create button for transition from current_sid to destination state.
       $button = array();
-      $button['#value'] = workflow_extensions_get_transition_label($form['#wf']->wid, $current_state_name, $to_state_name, $node);
+      $button['#value'] = workflow_extensions_get_transition_label($form['#wf']->wid, $current_state_name, workflow_get_state_name($sid), $node);
       $button['#type'] = 'submit';
       $button['#to_state'] = $sid;
       if (isset($form['buttons']['submit']['#weight'])) { // node form
@@ -436,4 +436,4 @@ function workflow_extensions_replace_tokens_raw($tokenized_label, $node = NULL)
     return token_replace_multiple($tokenized_label, $objects);
   }
   return $tokenized_label;
-}
\ No newline at end of file
+}