Updated migration