diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/metadata/OperatorGroupConstants.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/metadata/OperatorGroupConstants.scala
index 27ec48d4731..6adeffb71b8 100644
--- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/metadata/OperatorGroupConstants.scala
+++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/metadata/OperatorGroupConstants.scala
@@ -59,7 +59,12 @@ object OperatorGroupConstants {
     GroupInfo(SEARCH_GROUP),
     GroupInfo(
       CLEANING_GROUP,
-      List(GroupInfo(JOIN_GROUP), GroupInfo(AGGREGATE_GROUP), GroupInfo(SORT_GROUP))
+      List(
+        GroupInfo(JOIN_GROUP),
+        GroupInfo(SET_GROUP),
+        GroupInfo(AGGREGATE_GROUP),
+        GroupInfo(SORT_GROUP)
+      )
     ),
     GroupInfo(
       MACHINE_LEARNING_GROUP,