Fix bad method call in annotated state machine tests.