Fix problem in demo