Move the transformer management to the model