Fix migration II