Update to use new states flow