Fix bug in demo.