Add back the transition to/from edit mode