Fix failing Crag test