Updates to lakehouse migration