Add migration redo support