Updated migration to include old model