add initial state in stateful test example