Fix test states