Update state.py

reverting