Update schema after migration