diff --git a/src/cdk/tree/tree.ts b/src/cdk/tree/tree.ts
index 7deb63eb97dd..296e124245d2 100644
--- a/src/cdk/tree/tree.ts
+++ b/src/cdk/tree/tree.ts
@@ -399,16 +399,12 @@ export class CdkTree<T, K = T>
         this._nodeType,
         // NB: the data is unused below, however we add it here to essentially
         // trigger data rendering when expansion changes occur.
-        //
-        // TODO: remove debug code and comments
-        // Temporarily commenting this out to debug a test failure
-        //
-        // expansionModel.changed.pipe(
-        //   startWith(null),
-        //   tap(expansionChanges => {
-        //     this._emitExpansionChanges(expansionChanges);
-        //   }),
-        // ),
+        expansionModel.changed.pipe(
+          startWith(null),
+          tap(expansionChanges => {
+            this._emitExpansionChanges(expansionChanges);
+          }),
+        ),
       ])
         .pipe(
           switchMap(([data, nodeType]) => {