diff --git a/luamml-patches-mathtools.sty b/luamml-patches-mathtools.sty
index 3dc1570..76a9685 100644
--- a/luamml-patches-mathtools.sty
+++ b/luamml-patches-mathtools.sty
@@ -25,9 +25,9 @@
     \MH_fi:
     \ialign\bgroup
       \hfil\strut@$\m@th\displaystyle{}##
-      \luamml_save:nNn {} \displaystyle {mtd}
+      \UseTaggingSocket{math/luamml/save/nNn}{ {} \displaystyle {mtd}}
       $
-      \__luamml_amsmath_add_last_to_row:
+      \UseTaggingSocket{math/luamml/mtable/finalizecol}{last}
       \hfil
       \crcr
       \hfilneg