Fix test according to new exceptions