Fixing issues with the way I declared the FSM module