Fix state setup.
First passing test.