v2 patch (WIP)
v2 patch - fix bugs