Move migration tracking to workunit