Add support for updating only TF2 schema