diff --git a/workbench_moderation_notes.pages.inc b/workbench_moderation_notes.pages.inc
index 4de08d4..cb6b8ba 100644
--- a/workbench_moderation_notes.pages.inc
+++ b/workbench_moderation_notes.pages.inc
@@ -90,7 +90,7 @@ function workbench_moderation_notes_node_history_view($node) {
     if (!empty($node->workbench_moderation['published']) && $revision->vid == $node->workbench_moderation['published']->vid) {
       $row['data']['moderation'] = '<div class="moderation-state"><strong>' . t('This is the published revision.') . '</strong></div>';
       // Provide an unpublish link.
-      $next_states = workbench_moderation_states_next(workbench_moderation_state_published(), $user, $node->type);
+      $next_states = workbench_moderation_states_next(workbench_moderation_state_published(), $user, $node);
       if (!empty($next_states)) {
         $row['data']['moderation'] .= '<div class="moderation-actions">' . l(t('Unpublish'), "node/{$revision->nid}/moderation/{$revision->live_revision}/unpublish") . '</div>';
       }
@@ -150,4 +150,4 @@ function workbench_moderation_notes_node_history_view($node) {
     '#header' => $header,
     '#rows' => $rows,
   );
-}
\ No newline at end of file
+}