Add migration back