Fixed State.js usage in Edit.js