Adjust the demo code.