diff --git a/flytekit/models/execution.py b/flytekit/models/execution.py
index 4ab5d87104..40d58d83ac 100644
--- a/flytekit/models/execution.py
+++ b/flytekit/models/execution.py
@@ -342,6 +342,9 @@ def from_flyte_idl(cls, p):
             overwrite_cache=p.overwrite_cache,
             envs=_common_models.Envs.from_flyte_idl(p.envs) if p.HasField("envs") else None,
             tags=p.tags,
+            cluster_assignment=ClusterAssignment(cluster_pool=p.cluster_pool_name)
+            if p.HasField("cluster_pool_name")
+            else None,
         )