minor bug fix in demo