bug fixing, update test demo