Fix test after previous commit