Fix: use state.put for sanity