Two stage fallback failing and get stuck at action_listen

After user enter a message that activate two stage fallback. It get stuck at action_listen and the bot won’t respond anymore. Here is the error when I run locally:

and this is the error when I run on a server:

2022-08-08 03:46:25 ERROR    rasa.core.channels.telegram  - Exception when trying to handle message.'action_two_stage_fallback'
[2022-08-08 03:46:25 +0000] [1] [ERROR] Exception occurred while handling uri: '<my-url>'
Traceback (most recent call last):
  File "handle_request", line 83, in handle_request
    )
  File "/opt/venv/lib/python3.8/site-packages/rasa/core/channels/telegram.py", line 259, in message
    await on_new_message(
  File "/opt/venv/lib/python3.8/site-packages/rasa/core/channels/channel.py", line 89, in handler
    await app.ctx.agent.handle_message(message)
  File "/opt/venv/lib/python3.8/site-packages/rasa/core/agent.py", line 420, in handle_message
    return await self.processor.handle_message(  # type: ignore[union-attr]
  File "/opt/venv/lib/python3.8/site-packages/rasa/core/processor.py", line 147, in handle_message
    tracker = await self.run_action_extract_slots(message.output_channel, tracker)
  File "/opt/venv/lib/python3.8/site-packages/rasa/core/processor.py", line 173, in run_action_extract_slots
    extraction_events = await action_extract_slots.run(
  File "/opt/venv/lib/python3.8/site-packages/rasa/core/actions/action.py", line 1209, in run
    intent_is_desired = SlotMapping.intent_is_desired(
  File "/opt/venv/lib/python3.8/site-packages/rasa/shared/core/slot_mappings.py", line 112, in intent_is_desired
    + SlotMapping._get_active_loop_ignored_intents(
  File "/opt/venv/lib/python3.8/site-packages/rasa/shared/core/slot_mappings.py", line 93, in _get_active_loop_ignored_intents
    form_ignored_intents = domain.forms[active_loop_name].get(
KeyError: 'action_two_stage_fallback'

config.yml

  - name: FallbackClassifier
    threshold: 0.7
    ambiguity_threshold: 0.1
policies:
  - name: MemoizationPolicy
  - name: RulePolicy
    core_fallback_threshold: 0.3
    core_fallback_action_name: "action_default_fallback"
    enable_fallback_prediction: True
  - name: UnexpecTEDIntentPolicy
    max_history: 5
    epochs: 100
  - name: TEDPolicy
    max_history: 5
    epochs: 100
    constrain_similarities: true

domain.yml

  utter_ask_rephrase:
    - text: Sorry can you rephrase?
  utter_default:
    - text: Sorry I don't understand. Can you rephrase?

rules.yml

version: "3.1"
rules:
  - rule: Implementation of the Two-Stage-Fallback
    steps:
      - intent: nlu_fallback
      - action: action_two_stage_fallback
      - active_loop: action_two_stage_fallback
  - rule: Restart after a full two stage fallback
    condition:
      - active_loop: action_two_stage_fallback
    steps:
      - action: action_two_stage_fallback
      - active_loop: null
      - slot_was_set:
          - requested_slot: null
      - action: utter_default
      - action: action_default_fallback

I tried rules with and without the restart two stage rule