diff --git a/marimo/_plugins/ui/_impl/dataframes/transforms/apply.py b/marimo/_plugins/ui/_impl/dataframes/transforms/apply.py
index ca469c6f909..a5740ab754c 100644
--- a/marimo/_plugins/ui/_impl/dataframes/transforms/apply.py
+++ b/marimo/_plugins/ui/_impl/dataframes/transforms/apply.py
@@ -143,11 +143,9 @@ def _is_superset(self, transforms: Transformations) -> bool:
         if len(self._transforms) > len(transforms.transforms):
             return False
 
-        for i, transform in enumerate(self._transforms):
-            if transform != transforms.transforms[i]:
-                return False
-
-        return True
+        return (
+            self._transforms == transforms.transforms[: len(self._transforms)]
+        )
 
     def _get_next_transformations(
         self, transforms: Transformations