demo: fix small bug