Fix test broken by @agnogueira in previous commit