Bálint Junkuncz · Bálint Junkuncz
--- a/modules/custom_elements_ui/src/Form/EntityCustomElementsDisplayEditForm.php
+++ b/modules/custom_elements_ui/src/Form/EntityCustomElementsDisplayEditForm.php
 * @internal
 *   Form classes are internal.
 */
-class EntityCustomElementsDisplayEditForm extends EntityDisplayFormBase {
+final class EntityCustomElementsDisplayEditForm extends EntityDisplayFormBase {
--- a/src/Processor/DefaultContentEntityProcessor.php
+++ b/src/Processor/DefaultContentEntityProcessor.php
-    $this->entityTypeManager = $entityTypeManager;
-    return $this;
-  }
-
-  /**
-   * Gets the entity type manager.
-   *
-   * @return \Drupal\Core\Entity\EntityTypeManagerInterface
-   *   The entity type manager.
-   */
-  public function getEntityTypeManager() {
-    if (empty($this->entityTypeManager)) {
-      $this->entityTypeManager = \Drupal::entityTypeManager();
-    }
-    return $this->entityTypeManager;
+    $this->entityTypeManager = $entity_type_manager;
--- a/src/CustomElement.php
+++ b/src/CustomElement.php
   * @return $this
   */
  public function setSlotFromRenderArray($key, array $build, string $tag = 'div', array $attributes = [], int $index = NULL, int $weight = 0) {
-    $markup = \Drupal::service('renderer')->renderPlain($build);
+    $markup = $this->renderer->renderPlain($build);
--- a/src/CustomElementsConfigGenerator.php
+++ b/src/CustomElementsConfigGenerator.php
          $id = $entity_type_id . '.' . $bundle . '.default';
          $element_name = $entity_type_id . '-' . $bundle . '-default';

+          /** @var \Drupal\custom_elements\Entity\EntityCeDisplayInterface $config */
          $config = $storage->create([
            'id' => $id,
            'targetEntityType' => $entity_type_id,
            'bundle' => $bundle,
            'customElementName' => $element_name,
            'mode' => 'default',
+            'status' => TRUE,
          ]);

+          // These components must be set
+          // otherwise it breaks item generation.
+          $components = $config->getComponents();