Fix unit tests on travis