Another migration fix