adds correct state back in