diff --git a/features/update/basic/feature.py b/features/update/basic/feature.py
index 8a2f2650..dcd41f33 100644
--- a/features/update/basic/feature.py
+++ b/features/update/basic/feature.py
@@ -9,7 +9,8 @@
 
 @workflow.defn
 class Workflow:
-    am_done = False
+    def __init__(self) -> None:
+        self.am_done = False
 
     @workflow.run
     async def run(self) -> str:
diff --git a/features/update/self/feature.py b/features/update/self/feature.py
index 8fbacfa0..95d22978 100644
--- a/features/update/self/feature.py
+++ b/features/update/self/feature.py
@@ -9,7 +9,8 @@
 
 @workflow.defn
 class Workflow:
-    am_done = False
+    def __init__(self) -> None:
+        self.am_done = False
 
     @workflow.run
     async def run(self) -> str:
diff --git a/features/update/task_failure/feature.py b/features/update/task_failure/feature.py
index 736e353c..a596683f 100644
--- a/features/update/task_failure/feature.py
+++ b/features/update/task_failure/feature.py
@@ -12,7 +12,8 @@
 
 @workflow.defn
 class Workflow:
-    am_done = False
+    def __init__(self) -> None:
+        self.am_done = False
 
     @workflow.run
     async def run(self) -> str:
diff --git a/features/update/validation_replay/feature.py b/features/update/validation_replay/feature.py
index 34309aba..f44063f6 100644
--- a/features/update/validation_replay/feature.py
+++ b/features/update/validation_replay/feature.py
@@ -13,7 +13,8 @@
 
 @workflow.defn
 class Workflow:
-    am_done = False
+    def __init__(self) -> None:
+        self.am_done = False
 
     @workflow.run
     async def run(self) -> str: