fixing the issue in the demo