jetbrains-gateway 1.2.1 says agent_id not exist

module "jetbrains_gateway" {
  count  = data.coder_workspace.me.start_count
  source = "registry.coder.com/coder/jetbrains-gateway/coder"

  # JetBrains IDEs to make available for the user to select
  jetbrains_ides = ["PY"]
  default        = "PY"

  # Default folder to open when starting a JetBrains IDE
  folder = "/home/${local.username}/myproject"

  # This ensures that the latest non-breaking version of the module gets downloaded, you can also pin the module version to prevent breaking changes in production.
  version = "~> 1.2.1"

  agent_id   = coder_agent.main.id
  order      = 2
}

module "jetbrains_gateway" {
  count  = data.coder_workspace.me.start_count
  source = "registry.coder.com/coder/jetbrains-gateway/coder"

  # JetBrains IDEs to make available for the user to select
  jetbrains_ides = ["PY"]
  default        = "PY"

  # Default folder to open when starting a JetBrains IDE
  folder = "/home/${local.username}/myproject"

  # This ensures that the latest non-breaking version of the module gets downloaded, you can also pin the module version to prevent breaking changes in production.
  version = "~> 1.2.1"

  agent_id   = coder_agent.main.id
  order      = 2
}

resource "coder_agent" "main" {
  arch           = data.coder_provisioner.me.arch
  os             = "linux"
  startup_script = <<-EOT
    set -e

    # Prepare user home with default files on first start.
    if [ ! -f ~/.init_done ]; then
      cp -rT /etc/skel ~
      touch ~/.init_done
    fi

    # Add any commands that should be executed at workspace startup (e.g install requirements, start a program, etc) here
    
    ~/JetBrains/*/bin/remote-dev-server.sh registerBackendLocationForGateway
  EOT

...
}

resource "coder_agent" "main" {
  arch           = data.coder_provisioner.me.arch
  os             = "linux"
  startup_script = <<-EOT
    set -e

    # Prepare user home with default files on first start.
    if [ ! -f ~/.init_done ]; then
      cp -rT /etc/skel ~
      touch ~/.init_done
    fi

    # Add any commands that should be executed at workspace startup (e.g install requirements, start a program, etc) here
    
    ~/JetBrains/*/bin/remote-dev-server.sh registerBackendLocationForGateway
  EOT

...
}