fix: update theme migration logic to ensure completion before setting migration flag

Co-authored-by: Copilot <[email protected]>