Fix migration to work with current code.