diff --git a/hello/hello_search_attributes.py b/hello/hello_search_attributes.py
index 8b504ea6..14e5f39b 100644
--- a/hello/hello_search_attributes.py
+++ b/hello/hello_search_attributes.py
@@ -3,6 +3,10 @@
 from temporalio import workflow
 from temporalio.client import Client
 from temporalio.worker import Worker
+from temporalio.common import SearchAttributeKey
+
+# define our custom search attribute to modify later in the workflow
+custom_keyword_field = SearchAttributeKey.for_keyword("CustomKeywordField")
 
 
 @workflow.defn
@@ -11,7 +15,9 @@ class GreetingWorkflow:
     async def run(self) -> None:
         # Wait a couple seconds, then alter the keyword search attribute
         await asyncio.sleep(2)
-        workflow.upsert_search_attributes({"CustomKeywordField": ["new-value"]})
+        workflow.upsert_search_attributes([
+            custom_keyword_field.value_set("new-value")
+        ])
 
 
 async def main():
@@ -28,23 +34,24 @@ async def main():
         # While the worker is running, use the client to start the workflow.
         # Note, in many production setups, the client would be in a completely
         # separate process from the worker.
+
         handle = await client.start_workflow(
             GreetingWorkflow.run,
             id="hello-search-attributes-workflow-id",
             task_queue="hello-search-attributes-task-queue",
             # Start with default set of search attributes
-            search_attributes={"CustomKeywordField": ["old-value"]},
+            search_attributes=custom_keyword_field.value_set("old-value"),
         )
 
         # Show search attributes before and after a few seconds
         print(
             "First search attribute values: ",
-            (await handle.describe()).search_attributes.get("CustomKeywordField"),
+            (await handle.describe()).typed_search_attributes.get(custom_keyword_field),
         )
         await asyncio.sleep(3)
         print(
             "Second search attribute values: ",
-            (await handle.describe()).search_attributes.get("CustomKeywordField"),
+            (await handle.describe()).typed_search_attributes.get(custom_keyword_field),
         )