fix a demo issue