diff --git a/lib/Config/PostEditorBlock.php b/lib/Config/PostEditorBlock.php
index 2e73b82960..4612bb1466 100644
--- a/lib/Config/PostEditorBlock.php
+++ b/lib/Config/PostEditorBlock.php
@@ -36,7 +36,7 @@ class PostEditorBlock {
       $this->initFrontend();
     }
 
-    register_block_type('mailpoet/form-block-render', [
+    $this->wp->registerBlockType('mailpoet/form-block-render', [
       'attributes' => [
         'form' => [
           'type' => 'number',
@@ -63,12 +63,12 @@ class PostEditorBlock {
       Env::$version
     );
 
-    register_block_type('mailpoet/form-block', [
+    $this->wp->registerBlockType('mailpoet/form-block', [
       'style' => 'mailpoetblock-form-block-css',
       'editor_script' => 'mailpoet/form-block',
     ]);
 
-    add_action('admin_head', function() {
+    $this->wp->addAction('admin_head', function() {
       $forms = $this->formsRepository->findAllNotDeleted();
       ?>
       <script type="text/javascript">
@@ -84,7 +84,7 @@ class PostEditorBlock {
   }
 
   private function initFrontend() {
-    register_block_type('mailpoet/form-block', [
+    $this->wp->registerBlockType('mailpoet/form-block', [
       'render_callback' => [$this, 'renderForm'],
     ]);
   }
diff --git a/lib/WP/Functions.php b/lib/WP/Functions.php
index 824ee4d1ac..924d232ff9 100644
--- a/lib/WP/Functions.php
+++ b/lib/WP/Functions.php
@@ -468,6 +468,15 @@ class Functions {
     return wp_enqueue_style($handle, $src, $deps, $ver, $media);
   }
 
+  /**
+   * @param string|\WP_Block_Type $name
+   * @param array $args {
+   * @return \WP_Block_Type|false
+   */
+  public function registerBlockType($name, $args = []) {
+    return register_block_type($name, $args);
+  }
+
   public function wpGetAttachmentImageSrc($attachmentId, $size = 'thumbnail', $icon = false) {
     return wp_get_attachment_image_src($attachmentId, $size, $icon);
   }