Fix small demo.