diff --git a/projects/editor/src/app/components/unit-view/page-view/canvas/canvas-drag-overlay.component.ts b/projects/editor/src/app/components/unit-view/page-view/canvas/canvas-drag-overlay.component.ts
index f827ea0f6b252e69d48f062a5fb048bb4b9973e7..1f6f8156d151f97b84b80640d930460998e8cd4f 100644
--- a/projects/editor/src/app/components/unit-view/page-view/canvas/canvas-drag-overlay.component.ts
+++ b/projects/editor/src/app/components/unit-view/page-view/canvas/canvas-drag-overlay.component.ts
@@ -1,11 +1,8 @@
 import {
-  Component, OnInit, Input, Output,
-  EventEmitter,
-  ComponentFactoryResolver,
-  ViewChild, ViewContainerRef
+  Component, OnInit, Input, Output, EventEmitter, ComponentFactoryResolver, ViewChild, ViewContainerRef
 } from '@angular/core';
 import { UnitUIElement } from '../../../../../../../common/unit';
-import { CanvasElementComponent } from '../../../../../../../common/canvas-element-component.directive';
+import { FormElementComponent } from '../../../../../../../common/canvas-element-component.directive';
 import * as ComponentUtils from '../../../../../../../common/component-utils';
 
 @Component({
@@ -34,8 +31,9 @@ export class CanvasDragOverlayComponent implements OnInit {
   @Output() elementSelected = new EventEmitter<{
     componentElement: CanvasDragOverlayComponent,
     multiSelect: boolean }>();
+
   @ViewChild('elementContainer', { read: ViewContainerRef, static: true }) private elementContainer!: ViewContainerRef;
-  private childComponent!: CanvasElementComponent;
+  private childComponent!: FormElementComponent;
   _selected = false;
   style: Record<string, string> = {};
 
diff --git a/projects/player/src/app/components/element-overlay.component.ts b/projects/player/src/app/components/element-overlay.component.ts
index 453df03a5e0bc38d248e6bedd74cb7b6264e5491..6698086ba707e23dbaf3f47c0eac090beb1eebee 100644
--- a/projects/player/src/app/components/element-overlay.component.ts
+++ b/projects/player/src/app/components/element-overlay.component.ts
@@ -3,7 +3,7 @@ import {
 } from '@angular/core';
 import { FormGroup } from '@angular/forms';
 import { UnitUIElement } from '../../../../common/unit';
-import { CanvasElementComponent } from '../../../../common/canvas-element-component.directive';
+import { FormElementComponent } from '../../../../common/canvas-element-component.directive';
 import * as ComponentUtils from '../../../../common/component-utils';
 
 @Component({
@@ -24,9 +24,9 @@ export class ElementOverlayComponent implements OnInit {
   constructor(private componentFactoryResolver: ComponentFactoryResolver) { }
 
   ngOnInit(): void {
-    const componentFactory: ComponentFactory<CanvasElementComponent> =
+    const componentFactory: ComponentFactory<FormElementComponent> =
       ComponentUtils.getComponentFactory(this.element.type, this.componentFactoryResolver);
-    const childComponent: CanvasElementComponent =
+    const childComponent: FormElementComponent =
       this.elementContainer.createComponent(componentFactory).instance;
     childComponent.elementModel = this.element;
     childComponent.parentForm = this.parentForm;