Add state back