fix: fixed bug and update demo