Add migration test update

Extracted from #423