diff --git a/testing_action/glorious/pattern.py b/testing_action/glorious/pattern.py
index 9a72e13..c012e0b 100644
--- a/testing_action/glorious/pattern.py
+++ b/testing_action/glorious/pattern.py
@@ -1,4 +1,4 @@
-def rainbowify(string):
+def rainbowfier(string):
     colors = [
         "\033[91m",
         "\033[93m",
diff --git a/testing_action/main.py b/testing_action/main.py
index 8d8f5ce..978886d 100644
--- a/testing_action/main.py
+++ b/testing_action/main.py
@@ -1,7 +1,8 @@
-from glorious.pattern import rainbowify
+from glorious.pattern import rainbowfier
 
 from pyaction import PyAction
 from pyaction.workflow import annotations
+from pyaction.workflow.stream import WorkflowContext
 
 workflow = PyAction()
 
@@ -13,8 +14,10 @@ def testing_action(test_name: str, test_age: int) -> None:
     annotations.error("This is an error annotation!")
 
     workflow.write(
-        {
-            "message": rainbowify(f"{test_name}, you are {test_age}!"),
-            "multiline_message": f"{test_name=}\n{test_age=}",
-        }
+        WorkflowContext(
+            {
+                "message": rainbowfier(f"{test_name}, you are {test_age}!"),
+                "multiline_message": f"{test_name=}\n{test_age=}",
+            }
+        )
     )