Fix small problems on running the demo code

Signed-off-by: abelxie <jinjing.xie@gmail.com>