Add schematic migration support