Fix tests

Fix test on buildbot