diff --git a/src/demo-app/tree/checklist-tree-demo/checklist-tree-demo.ts b/src/demo-app/tree/checklist-tree-demo/checklist-tree-demo.ts
index 69f632e9f380..a7790ce6c881 100644
--- a/src/demo-app/tree/checklist-tree-demo/checklist-tree-demo.ts
+++ b/src/demo-app/tree/checklist-tree-demo/checklist-tree-demo.ts
@@ -10,6 +10,7 @@ import {SelectionModel} from '@angular/cdk/collections';
 import {FlatTreeControl} from '@angular/cdk/tree';
 import {MatTreeFlattener, MatTreeFlatDataSource} from '@angular/material/tree';
 import {TodoItemNode, ChecklistDatabase} from './checklist-database';
+import {BehaviorSubject} from 'rxjs';
 
 /**
  * Checklist demo with flat tree
@@ -51,7 +52,7 @@ export class ChecklistTreeDemo {
     return node.children.value.length > 0;
   }
 
-  getChildren = (node: TodoItemNode) => {
+  getChildren = (node: TodoItemNode): BehaviorSubject<TodoItemNode[]> => {
     return node.children;
   }