diff --git a/src/Form/LayoutBuilderForm.php b/src/Form/LayoutBuilderForm.php
index b60a4ec..773df3e 100644
--- a/src/Form/LayoutBuilderForm.php
+++ b/src/Form/LayoutBuilderForm.php
@@ -4,6 +4,7 @@ namespace Drupal\layout_library\Form;
 
 use Drupal\Core\Entity\EntityForm;
 use Drupal\Core\Form\FormStateInterface;
+use Drupal\layout_builder\Form\PreviewToggleTrait;
 use Drupal\layout_builder\LayoutTempstoreRepositoryInterface;
 use Drupal\layout_builder\SectionStorageInterface;
 use Symfony\Component\DependencyInjection\ContainerInterface;
@@ -15,6 +16,8 @@ use Symfony\Component\DependencyInjection\ContainerInterface;
  */
 class LayoutBuilderForm extends EntityForm {
 
+  use PreviewToggleTrait;
+
   /**
    * Layout tempstore repository.
    *
@@ -75,6 +78,7 @@ class LayoutBuilderForm extends EntityForm {
       '#attributes' => ['class' => ['button']],
       '#url' => $this->sectionStorage->getLayoutBuilderUrl('discard_changes'),
     ];
+    $actions['preview_toggle'] = $this->buildContentPreviewToggle();
     return $actions;
   }