tests: fix remaining unit tests