Fix test that was not updated.