Fix unit tests on Ubuntu