diff --git a/publishcontent.module b/publishcontent.module
index b87b4e5..c951875 100644
--- a/publishcontent.module
+++ b/publishcontent.module
@@ -19,6 +19,7 @@ function publishcontent_menu() {
     'access arguments' => array(1, 3),
     'weight' => 5,
     'type' => MENU_LOCAL_TASK,
+    'options' => array('attributes' => array('class' => array('publish'))),
   );
   $items['node/%node/unpublish/%publishcontent_security_token'] = array(
     'title' => 'Unpublish',
@@ -28,6 +29,7 @@ function publishcontent_menu() {
     'access arguments' => array(1, 3),
     'weight' => 5,
     'type' => MENU_LOCAL_TASK,
+    'options' => array('attributes' => array('class' => array('unpublish'))),
   );
   return $items;
 }
diff --git a/publishcontent_views_handler_field_node_link.inc b/publishcontent_views_handler_field_node_link.inc
index 4111826..ef5cfc6 100644
--- a/publishcontent_views_handler_field_node_link.inc
+++ b/publishcontent_views_handler_field_node_link.inc
@@ -48,14 +48,14 @@ class publishcontent_views_handler_field_node_link extends views_handler_field {
     $node->status = $values->{$this->aliases['status']};
     $node->uid = $values->{$this->aliases['uid']};
 
-    $options = array('query' => drupal_get_destination());
-
     if ($node->status && _publishcontent_unpublish_access($node))
       $op = 'unpublish';
 
     if (!$node->status && _publishcontent_publish_access($node))
       $op = 'publish';
 
+    $options = array('query' => drupal_get_destination(), 'attributes' => array('class' => $op));
+
     if (isset($op))
       return l(empty($this->options[$op]) ? t($op) : $this->options[$op],
                "node/$node->nid/$op/". drupal_get_token(), $options);