diff --git a/src/Form/LayoutBuilderForm.php b/src/Form/LayoutBuilderForm.php
index 86ece50..f85501f 100644
--- a/src/Form/LayoutBuilderForm.php
+++ b/src/Form/LayoutBuilderForm.php
@@ -2,18 +2,12 @@
 
 namespace Drupal\page_manager\Form;
 
-use Drupal\Core\Entity\EntityForm;
-use Drupal\Core\Entity\EntityType;
-use Drupal\Core\Entity\EntityTypeManager;
 use Drupal\Core\Form\FormBase;
 use Drupal\Core\Form\FormStateInterface;
 use Drupal\Core\Plugin\Context\EntityContext;
-use Drupal\Core\Routing\RouteMatchInterface;
-use Drupal\ctools\Wizard\FormWizardBase;
 use Drupal\layout_builder\Form\PreviewToggleTrait;
 use Drupal\layout_builder\LayoutTempstoreRepositoryInterface;
 use Drupal\layout_builder\SectionStorage\SectionStorageManager;
-use Drupal\layout_builder\SectionStorageInterface;
 use Symfony\Component\DependencyInjection\ContainerInterface;
 
 /**
@@ -99,12 +93,17 @@ class LayoutBuilderForm extends FormBase {
    * {@inheritdoc}
    */
   public function submitForm(array &$form, FormStateInterface $form_state) {
+    $this->sectionStorage->save();
+
+    // \Drupal\ctools\Wizard\EntityFormWizardBase::finish will fetch the page
+    // from the form temporary values. Let's update that with the new values.
+    $cached_values = $form_state->getTemporaryValue('wizard');
+    // Override with the updated variant.
+    $cached_values['page']->addVariant($this->sectionStorage->getContextValue('page_variant'));
+    $form_state->setTemporaryValue('wizard', $cached_values);
 
-    $return = $this->sectionStorage->save();
     $this->layoutTempstoreRepository->delete($this->sectionStorage);
     $this->messenger()->addMessage($this->t('The layout has been saved.'));
-    $form_state->setRedirectUrl($this->sectionStorage->getRedirectUrl());
-    return $return;
   }
 
   /**
diff --git a/src/Plugin/DisplayVariant/LayoutBuilderDisplayVariant.php b/src/Plugin/DisplayVariant/LayoutBuilderDisplayVariant.php
index bae01ae..0493b78 100644
--- a/src/Plugin/DisplayVariant/LayoutBuilderDisplayVariant.php
+++ b/src/Plugin/DisplayVariant/LayoutBuilderDisplayVariant.php
@@ -4,15 +4,9 @@ namespace Drupal\page_manager\Plugin\DisplayVariant;
 
 use Drupal\Core\Display\VariantBase;
 use Drupal\Core\Form\FormStateInterface;
-use Drupal\Core\Plugin\Context\EntityContext;
-use Drupal\Core\Plugin\PluginFormInterface;
 use Drupal\ctools\Plugin\PluginWizardInterface;
-use Drupal\layout_builder\Form\DefaultsEntityForm;
 use Drupal\layout_builder\SectionStorage\SectionStorageTrait;
 use Drupal\page_manager\Form\LayoutBuilderForm;
-use Drupal\panels\Form\LayoutChangeRegions;
-use Drupal\panels\Form\LayoutChangeSettings;
-use Drupal\panels\Form\LayoutPluginSelector;
 
 /**
  * Provides a Layout Builder variant.
@@ -30,7 +24,11 @@ class LayoutBuilderDisplayVariant extends VariantBase implements PluginWizardInt
    * {@inheritdoc}
    */
   public function build() {
-    return ['#markup' => 'Hello'];
+    $build = [];
+    foreach ($this->getSections() as $delta => $section) {
+      $build[$delta] = $section->toRenderArray();
+    }
+    return $build;
   }
 
   public function buildConfigurationForm(array $form, FormStateInterface $form_state) {