Update test utils to work with the state event updates