Update scheme after migration