diff --git a/src/Plugin/block_field/BlockFieldSelection/Blocks.php b/src/Plugin/block_field/BlockFieldSelection/Blocks.php
index 045f352..aced345 100644
--- a/src/Plugin/block_field/BlockFieldSelection/Blocks.php
+++ b/src/Plugin/block_field/BlockFieldSelection/Blocks.php
@@ -3,6 +3,7 @@
 namespace Drupal\block_field\Plugin\block_field\BlockFieldSelection;
 
 use Drupal\block_field\BlockFieldSelectionBase;
+use Drupal\Core\DependencyInjection\DependencySerializationTrait;
 use Drupal\Core\Form\FormStateInterface;
 use Drupal\Core\StringTranslation\StringTranslationTrait;
 
@@ -15,6 +16,8 @@ use Drupal\Core\StringTranslation\StringTranslationTrait;
  * )
  */
 class Blocks extends BlockFieldSelectionBase {
+
+  use DependencySerializationTrait;
   use StringTranslationTrait;
 
   /**
diff --git a/src/Plugin/block_field/BlockFieldSelection/Categories.php b/src/Plugin/block_field/BlockFieldSelection/Categories.php
old mode 100644
new mode 100755
index 771d276..f01c0b8
--- a/src/Plugin/block_field/BlockFieldSelection/Categories.php
+++ b/src/Plugin/block_field/BlockFieldSelection/Categories.php
@@ -3,6 +3,7 @@
 namespace Drupal\block_field\Plugin\block_field\BlockFieldSelection;
 
 use Drupal\block_field\BlockFieldSelectionBase;
+use Drupal\Core\DependencyInjection\DependencySerializationTrait;
 use Drupal\Core\Form\FormStateInterface;
 use Drupal\Core\StringTranslation\StringTranslationTrait;
 
@@ -15,6 +16,8 @@ use Drupal\Core\StringTranslation\StringTranslationTrait;
  * )
  */
 class Categories extends BlockFieldSelectionBase {
+
+  use DependencySerializationTrait;
   use StringTranslationTrait;
 
   /**