Fix failing test in 4c19e98