Fixed return type on the transformer