I see.. I agree the answer is in the `TemplateCompilationUnit` camp, and it currently doesn't support the two modes of operation. I am very reluctant to global flags, but I don't want to block this great improvement for too long.

Please add a `FIXME` comment here, and open a ticket describing the issue, and we'll tackle that in the next sprint.