Add back removed migration