Fix demo

Fix demo