test: Improve TestState3