Quick migration fix