diff --git a/src/framework/theme/components/tree-grid/tree-grid.component.ts b/src/framework/theme/components/tree-grid/tree-grid.component.ts
index 63e6ae9731..c3e2ae6992 100644
--- a/src/framework/theme/components/tree-grid/tree-grid.component.ts
+++ b/src/framework/theme/components/tree-grid/tree-grid.component.ts
@@ -21,6 +21,7 @@ import {
   Optional,
   SkipSelf,
 } from '@angular/core';
+import { CDK_TABLE } from '@angular/cdk/table';
 import { fromEvent, merge, Subject } from 'rxjs';
 import { debounceTime, takeUntil } from 'rxjs/operators';
 
@@ -143,6 +144,7 @@ import { NbColumnsService } from './tree-grid-columns.service';
   styleUrls: ['./tree-grid.component.scss'],
   providers: [
     { provide: NB_TREE_GRID, useExisting: NbTreeGridComponent },
+    { provide: CDK_TABLE, useExisting: NbTreeGridComponent },
     NbColumnsService,
     ...NB_TABLE_PROVIDERS,
   ],