Fixes compile errors in migrator