Add schema changes to migration fix