Fix test regression introduced by 976ba90