Fix errors in demo