Fix tests on ci