fix demo's bug