State generation should work now. See in code comments.