Atualização: As alterações neste fork foram mescladas no upstream, por favor, use os componentes de tema oficiais em vez disso
Para quem quiser aplicar o DiscoTOC em suas respostas, eu fiz essa parte do trabalho no meu próprio fork!
Agora ele pode aplicar o TOC a qualquer postagem e funciona bem, assim como faz no OP.
Parece que a equipe oficial não acha necessário aplicar o DiscoTOC nas respostas, então não abrirei um pull request. Mas você pode usar minha versão modificada!