Fix output connection of Transformer