Fix demo scripts