Fix the bug on demo page