fix migration generator