diff --git a/tests/cluster_policies/__init__.py b/tests/cluster_policies/__init__.py
index b54667647c5cf..dbb5d0163dc8a 100644
--- a/tests/cluster_policies/__init__.py
+++ b/tests/cluster_policies/__init__.py
@@ -82,15 +82,15 @@ def dag_policy(dag: DAG):
 
 
 # [START example_task_cluster_policy]
-class TimedOperator(BaseOperator, ABC):
-    timeout: timedelta
+class MyCustomOperator(BaseOperator, ABC):
+    pass
 
 
-def task_policy(task: TimedOperator):
-    if task.task_type == 'HivePartitionSensor':
-        task.queue = "sensor_queue"
-    if task.timeout > timedelta(hours=48):
-        task.timeout = timedelta(hours=48)
+def task_policy(task: MyCustomOperator):
+    if task.task_type == 'BaseOperator':
+        task.queue = "priority_queue"
+    if task.execution_timeout > timedelta(hours=48):
+        task.execution_timeout = timedelta(hours=48)
 
 
 # [END example_task_cluster_policy]