Update the lib to match the last updates and add unit tests

### Describe the problem you discovered

Make sure the lib is up to date with the latest changes on the node side

### Describe the solution you'd like

Update the code and write corresponding unit tests

Start by comparing the Go lib with the latest changes on the node side, since the issue does not name specific files or entry points. Identify which library behavior is out of date, then update the corresponding code and add unit tests that cover those changes. Done means the lib matches the node-side updates and the tests pass.

Written by the indexing model from the issue text.