Add in updated schema created after migration