fixing bug in demo
fixing bug in demo