fix demo issue