diff --git a/src/Plugin/ExtraField/Display/LeaveReviewButton.php b/src/Plugin/ExtraField/Display/LeaveReviewButton.php
index 3e8bcb0ef6eff9f30601b21f7c69cd9f261833bf..e0dbf77e226ec723016cdc236e964c05156060bc 100644
--- a/src/Plugin/ExtraField/Display/LeaveReviewButton.php
+++ b/src/Plugin/ExtraField/Display/LeaveReviewButton.php
@@ -37,12 +37,13 @@ class CreateReferencingContentButton extends ExtraFieldPlusDisplayBase {
 
     $url = base_path() . 'node/add/pece_annotation?edit[field_annotation_artifact][widget][0][target_id]=' . $entity->id();
     $build = [
-      '#type' => 'inline_template',
-      '#template' => '',
-      '#context' => [
+      '#type' => 'component',
+      '#component' => 'create_referencing_content:create-referencing-content-button',
+      '#props' => [
         'url' => $url,
         'label' => $this->t('Annotate'),
         'tooltip' => $this->t('Create new Annotation content referencing this :content_type', [':content_type' => $content_type]),
+        'classes' => ['button is-primary is-medium'],
       ],
     ];
     return $build;