small code change in demo