Remove unneeded state persistence for demo