Fix the broken demo script