[Refactor] Update util functions to properly use `my-avbroot-setup`

see: https://github.com/chenxiaolong/my-avbroot-setup/pull/6 i feel, this is some sort of a breaking change, so will have to go through the changes and make changes accordingly.