diff --git a/python/ray/data/_internal/logical/interfaces/optimizer.py b/python/ray/data/_internal/logical/interfaces/optimizer.py
index 70b235b42241..6a25a44afe62 100644
--- a/python/ray/data/_internal/logical/interfaces/optimizer.py
+++ b/python/ray/data/_internal/logical/interfaces/optimizer.py
@@ -6,7 +6,7 @@
 class Rule:
     """Abstract class for optimization rule."""
 
-    def apply(plan: Plan) -> Plan:
+    def apply(self, plan: Plan) -> Plan:
         """Apply the optimization rule to the execution plan."""
         raise NotImplementedError