I can't view the screenshot for some unknown reason.

Rather than adding a step, look in the Applied Steps of the 'final' query. You should see the following steps: Invoke Function, Renamed Columns, Removed Columns and Expanded Table.

Edit the "Removed Columns" step so that the columns you wish to keep are not removed.