UIPath Acadamy DU Practice Exercise - Build an end-to-end Document Understanding

I went through the solution given for Document Understanding Framework Practice exercise in UiPath Acadamy. I also tried to run it in debug mode. I see the classifiers are classifying the input file only in four files.

First three pages in the input file are of type W9, current code is grouping all three pages in single group and code is extracting data only from page 1 in further steps, but ideally it should extract data from all three pages. How do we fix the issue?