Fix broken migration