diff --git a/subiquity/common/types.py b/subiquity/common/types.py
index 5155a0e1..b0382e5d 100644
--- a/subiquity/common/types.py
+++ b/subiquity/common/types.py
@@ -819,9 +819,11 @@ class TaskStatus(enum.Enum):
     DO = "Do"
     DOING = "Doing"
     DONE = "Done"
+    WAIT = "Wait"
     ABORT = "Abort"
     UNDO = "Undo"
     UNDOING = "Undoing"
+    UNDONE = "Undone"
     HOLD = "Hold"
     ERROR = "Error"