State seems to be working. Tests pass