Remove protected for aasm_write_state