fixed an error in the demo