diff --git a/luamml-patches-lab-math.sty b/luamml-patches-lab-math.sty
index f329911..c74a206 100644
--- a/luamml-patches-lab-math.sty
+++ b/luamml-patches-lab-math.sty
@@ -1,9 +1,13 @@
 \ProvidesExplPackage {luamml-patches-lab-math} {2024-10-30} {0.2.0}
   {Feel free to add a description here}
 
-% This definition is identical to the one in latex-lab-math. 
+% This definition is identical to the one in latex-lab-math.
 % The redefinition and the whole patch file can be removed in 2025-06-01
-\AddToHook{begindocument} {
+\IfPackageAtLeastTF{latex-lab-testphase-math}{2025-01-24}
+ {}
+ {
+  \AddToHook{begindocument} {
+  \PackageInfo{luamml}{patching~\string\common@align@ending}
   \cs_set:Npn \common@align@ending {
     \math@cr
     \black@ \totwidth@
@@ -18,5 +22,6 @@
       $$
     \fi
     \ignorespacesafterend
+   }
   }
 }