Fixed schema to work with updates ui