From c472a745a1e1db243f2b6d367a7364735e380a83 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michal=20Tich=C3=A1k?= <michal.tichak@cern.ch>
Date: Fri, 8 May 2026 16:05:38 +0200
Subject: [PATCH] [core] Hook control mode is no longer changed to direct

---
 core/task/taskclass/class.go | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/core/task/taskclass/class.go b/core/task/taskclass/class.go
index 902741c6c..0acf88a93 100644
--- a/core/task/taskclass/class.go
+++ b/core/task/taskclass/class.go
@@ -123,7 +123,6 @@ func (c *Class) UnmarshalYAML(unmarshal func(interface{}) error) (err error) {
 		}
 	}
 	return
-
 }
 
 func (c *Class) MarshalYAML() (interface{}, error) {
@@ -153,14 +152,7 @@ func (c *Class) MarshalYAML() (interface{}, error) {
 		Constraints: c.Constraints,
 		Command:     c.Command,
 	}
-
-	if c.Control.Mode == controlmode.FAIRMQ {
-		aux.Control.Mode = "fairmq"
-	} else if c.Control.Mode == controlmode.BASIC {
-		aux.Control.Mode = "basic"
-	} else {
-		aux.Control.Mode = "direct"
-	}
+	aux.Control.Mode = c.Control.Mode.String()
 
 	return aux, nil
 }