Add #preserve_state