Fix tests on CI