Add conditional migration