Changing the return type in typescript / Lit breaks treesitter capture

[aaaaargZombies]

No return type specified, works as expect. But this code breaks the capture.

    const remoteDataSwitch = (rd: RemoteData): TemplateResult => {
      switch (rd.kind) {
        case 'Loading':
          return html`<p>LOADING STATE</p>`;
        case 'NotAsked':
          return html`<p>NOT_ASKED STATE</p>`;
        case 'Success':
          return html`<p>SUCCESS STATE</p>
            <p>
              <code>${rd.result.getChannelData(0).length}</code>
            </p>`;
        case 'Failure':
          return html`<p>FAILURE STATE</p>`;
      }

Though it seems to work in treesitter playground, which I now see is archived? could be a potential issue.