Fix broken migration. OOPS