lb: fix unit tests after rebase