[migration] Fix migration