Fix demo script