diff --git a/deploy-agent/deployd/types/deploy_goal.py b/deploy-agent/deployd/types/deploy_goal.py
index bd08c89661..906f6d6b04 100644
--- a/deploy-agent/deployd/types/deploy_goal.py
+++ b/deploy-agent/deployd/types/deploy_goal.py
@@ -35,7 +35,7 @@ def __init__(self, jsonValue=None):
             self.envId = jsonValue.get('envId')
             self.envName = jsonValue.get('envName')
             self.stageName = jsonValue.get('stageName')
-            self.stageName = jsonValue.get('stageType')
+            self.stageType = jsonValue.get('stageType')
             # TODO: Only used for migration, should remove later
             if isinstance(jsonValue.get('deployStage'), int):
                 self.deployStage = DeployStage._VALUES_TO_NAMES[jsonValue.get('deployStage')]