Fix migration generator