diff --git a/projects/plotly/src/lib/plotly.component.ts b/projects/plotly/src/lib/plotly.component.ts
index 9940288..3b1da15 100644
--- a/projects/plotly/src/lib/plotly.component.ts
+++ b/projects/plotly/src/lib/plotly.component.ts
@@ -25,7 +25,9 @@ import { Plotly } from './plotly.interface';
 // @dynamic
 @Component({
     selector: 'plotly-plot',
-    template: `<div #plot [attr.id]="divId" [className]="getClassName()" [ngStyle]="style"></div>`,
+    template: `<div #plot [attr.id]="divId" [ngClass]="getClassName()" [ngStyle]="style">
+      <ng-content></ng-content>
+    </div>`,
     providers: [PlotlyService],
 })
 export class PlotlyComponent implements OnInit, OnChanges, OnDestroy, DoCheck {