diff --git a/sharethis.module b/sharethis.module
index 9b45c1c..3880103 100644
--- a/sharethis.module
+++ b/sharethis.module
@@ -106,7 +106,7 @@ function sharethis_node_view(array &$build, EntityInterface $node, EntityViewDis
       if (isset($enabled_types[$node->bundle()]) && $enabled_types[$node->bundle()] === $node->bundle() && $display->getcomponent('sharethis')) {
         $st_js = $sharethis_manager->sharethisIncludeJs();
         $content = $sharethis_manager->renderSpans($data_options, $title, $path);
-        $build['sharethis_content'] = array(
+        $build['sharethis'] = array(
           '#theme' => 'sharethis_block',
           '#content' => $content,
           '#attached' => array(
@@ -185,7 +185,7 @@ function sharethis_comment_view(array &$build, EntityInterface $comment, EntityV
     $route_match = \Drupal::routeMatch();
     $title = \Drupal::service('title_resolver')->getTitle($request, $route_match->getRouteObject());
     $content = $sharethis_manager->renderSpans($data_options, $title, $path);
-    $build['sharethis_comments'] = array(
+    $build['sharethis'] = array(
       '#theme' => 'sharethis_block',
       '#content' => $content,
       '#attached' => array(
@@ -198,7 +198,7 @@ function sharethis_comment_view(array &$build, EntityInterface $comment, EntityV
           'sharethis' => $st_js,
         ),
       ),
-      '#attributes' => array('class' => 'sharethis-comment'),
+      '#attributes' => array('class' => 'sharethis'),
       '#weight' => $sharethis_settings->get('location'),
     );
   }