Add back in the transitions