Add some missing keys in the schema