Fix failing crag test