diff --git a/numaflow/src/shared.rs b/numaflow/src/shared.rs
index 7d9ecab..a940f8a 100644
--- a/numaflow/src/shared.rs
+++ b/numaflow/src/shared.rs
@@ -78,7 +78,10 @@ pub(crate) struct ServerInfo {
 impl ServerInfo {
     pub fn new(container_type: ContainerType) -> Self {
         let mut metadata: HashMap<String, String> = HashMap::new();
-        if container_type == ContainerType::Map || container_type == ContainerType::BatchMap {
+        if container_type == ContainerType::Map
+            || container_type == ContainerType::BatchMap
+            || container_type == ContainerType::MapStream
+        {
             metadata.insert(
                 MAP_MODE_KEY.to_string(),
                 match container_type {