diff --git a/src/gateway/session.rs b/src/gateway/session.rs
index c7784d4..7bc6877 100644
--- a/src/gateway/session.rs
+++ b/src/gateway/session.rs
@@ -950,6 +950,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1001,6 +1002,7 @@ mod tests {
             ]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1068,6 +1070,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1152,6 +1155,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1237,6 +1241,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1321,6 +1326,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1387,6 +1393,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1462,6 +1469,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )
@@ -1524,6 +1532,7 @@ mod tests {
             HashMap::from([("default".to_string(), provider_config)]),
             Arc::new(SkillRuntime::default()),
             HashSet::new(),
+            crate::config::TaskConfig::default(),
             Some(4),
             Some(24),
         )