Are you sure this doesn't cause us to lose the stageId column? I thought the DSL requires that all Scala be preceded by brackets. Same for in headers. (It does seem to be gone in the screenshot you showed.)