Fix unit tests on ubuntu