Fix test after a3ab721ac6e2769b08621ed8189f7636cf403a0d