Fix gen_fsm example code