Schema fix

Fix to the shema