trivial: Fix unit tests