Fix demo update fix