fix(demo): improve it