Fix tests on CI. Cry tears of shame.