update small fix for demo