Add forgotten changes to ModelDrivenTransform