This example is failing on CI, but passes locally. In #171, I commented out the assert that was failing. We should uncomment it once we work out why it fails
