diff --git a/projects/swimlane/ngx-datatable/src/lib/components/header/header-cell.component.ts b/projects/swimlane/ngx-datatable/src/lib/components/header/header-cell.component.ts
index 4ec70918e..ed86935e3 100644
--- a/projects/swimlane/ngx-datatable/src/lib/components/header/header-cell.component.ts
+++ b/projects/swimlane/ngx-datatable/src/lib/components/header/header-cell.component.ts
@@ -64,7 +64,7 @@ export class DataTableHeaderCellComponent {
     return this._allRowsSelected;
   }
 
-  @Input() selectionType: SelectionType;
+  @Input() selectionType: SelectionType | keyof typeof SelectionType;
 
   @Input() set column(column: TableColumn) {
     this._column = column;