The test example needs to be updated as well :smile: