[BUG] Numbering of restated theorems is not the same as in the original theorem using jmlr

[jcalifornia]

Describe the bug
The numbering doesn't reset. See the example below where it is Lemma 1,2 in main text and is Lemma 3,4 in the appendix when restated

To Reproduce: Minimal Working Example
Simple example:

\documentclass[pmlr]{jmlr}% new name PMLR (Proceedings of Machine Learning)
\usepackage[createShortEnv]{proof-at-the-end}


\begin{document}


\begin{lemmaE}[Something here][end,restate,text link section]
\end{lemmaE}
\begin{proofE}
The proof is self-evident
\end{proofE}

\begin{lemmaE}[Something else here][end,restate,text link section]
\end{lemmaE}
\begin{proofE}
The proof is also self-evident
\end{proofE}

\appendix


\section{Proofs}

\printProofs

\end{document}

The ouput is:

Lemma 1 (Something here)
See proof in section A .
Lemma 2 (Something else here)
See proof in section A .
Appendix A. Proofs
Lemma 3 (Something here)
Proof [Proof of Theorem 1]The proof is self-evident
Lemma 4 (Something else here)
Proof [Proof of Theorem 2]The proof is also self-evident

Version
Overleaf texlive 2022

Version of proof-at-the-end: master

[jcalifornia]

Works if I define my own counters

[dpo]

Could you please show how you solved the problem?

[jcalifornia]

I added these to the start of the appendix

\setcounter{lemma}{0}
\setcounter{theorem}{0}
\setcounter{equation}{0}

[tobiasBora]

Note that you might want to use cref instead that is more resilient than the default autoref, and add section to the counters taken into account by thm-restate as explained here #14 (comment)