Cannot write persist image Xiaomi Mi6

Trying to fix the home button problems in my xiaomi MI6, I'm trying to flash a complete persist img from another friends Mi6. However, the command ends:

python3.6 edl.py w persist dumps/persist.img --memory=ufs --lun=0 --loader Loaders/xiaomi/prog_ufs_firehose_8998_ddr.elf
Capstone library is missing (optional).
Keystone library is missing (optional).
Qualcomm Sahara / Firehose Client V3.52 (c) B.Kerler 2018-2021.
main - Using loader Loaders/xiaomi/prog_ufs_firehose_8998_ddr.elf ...
main - Waiting for the device
main - Device detected :)
main - Mode detected: firehose
firehose - Chip serial num: 0 (0x0)
firehose - Supported Functions: demacia,setprojmodel
firehose -
firehose - TargetName=MSM8996
firehose - MemoryName=UFS
firehose - Version=1
firehose_client - Supported functions:
-----------------
demacia,setprojmodel
firehose -
Writing to physical partition 0, sector 16384, sectors 16384
Traceback (most recent call last):
  File "edl.py", line 362, in <module>
    base.run()
  File "edl.py", line 344, in run
    fh.handle_firehose(cmd, options)
  File "/Users/charlyo/edl/edl/Library/firehose_client.py", line 651, in handle_firehose
    if self.firehose.cmd_program(lun, startsector, filename):
  File "/Users/charlyo/edl/edl/Library/firehose.py", line 440, in cmd_program
    data += self.modules.addprogram()
  File "/Users/charlyo/edl/edl/Library/Modules/init.py", line 64, in addprogram
    return self.ops.addprogram()
  File "/Users/charlyo/edl/edl/Library/Modules/oneplus.py", line 239, in addprogram
    pk, token = self.ops.generatetoken(True)
AttributeError: 'NoneType' object has no attribute 'generatetoken'

Any idea on how to fix this? Persist image is in lun=0 for my device.

bkerler / edl

Cannot write persist image Xiaomi Mi6 #166