Fix the bug in demo

Since you change some variable name in the project , the demo can't be run. I make a little change so the demo can be ok now.