migration generator fix