diff --git a/src/Plugin/Field/FieldFormatter/VoteUpDownFormatterType.php b/src/Plugin/Field/FieldFormatter/VoteUpDownFormatterType.php
index 3c73eca..006c46d 100644
--- a/src/Plugin/Field/FieldFormatter/VoteUpDownFormatterType.php
+++ b/src/Plugin/Field/FieldFormatter/VoteUpDownFormatterType.php
@@ -2,8 +2,11 @@
 
 namespace Drupal\vud\Plugin\Field\FieldFormatter;
 
+use Drupal\Component\Utility\Html;
+use Drupal\Core\Field\FieldItemInterface;
 use Drupal\Core\Field\FieldItemListInterface;
 use Drupal\Core\Field\FormatterBase;
+use Drupal\Core\Form\FormStateInterface;
 
 /**
  * Plugin implementation of the 'vote_up_down_formatter_type' formatter.
@@ -22,13 +25,11 @@ class VoteUpDownFormatterType extends FormatterBase {
    * {@inheritdoc}
    */
   public function viewElements(FieldItemListInterface $items, $langcode) {
-    $elements = [];
 
     $widget = \Drupal::service('plugin.manager.vud')
       ->createInstance($this->getFieldSetting('widget'));
 
-    $element = $widget->build($items->getEntity());
-    $element['items'] = [];
+    $element[] = $widget->build($items->getEntity());
 
     return $element;
   }