Fix broken unit tests on CI machine