Fix bugs for demo_hero modules