Added comfy state migration