diff --git a/CHANGELOG.md b/CHANGELOG.md
index 48220e21..879f33dd 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -15,6 +15,12 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ### Fixed
 
+
+
+# [1.14.1](https://github.com/cludden/protoc-gen-go-temporal/releases/tag/v1.14.1) - 2024-06-20
+
+### Fixed
+
 - [#73](https://github.com/cludden/protoc-gen-go-temporal/pull/73) default to WorkflowUpdateStageCompleted if update options WaitForStage unspecified
 
 
diff --git a/gen/example/helloworld/v1/example.pb.go b/gen/example/helloworld/v1/example.pb.go
index 7bc459b9..d2fbb084 100644
--- a/gen/example/helloworld/v1/example.pb.go
+++ b/gen/example/helloworld/v1/example.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: example/helloworld/v1/example.proto
 
@@ -230,7 +230,7 @@ func file_example_helloworld_v1_example_proto_rawDescGZIP() []byte {
 }
 
 var file_example_helloworld_v1_example_proto_msgTypes = make([]protoimpl.MessageInfo, 3)
-var file_example_helloworld_v1_example_proto_goTypes = []interface{}{
+var file_example_helloworld_v1_example_proto_goTypes = []any{
 	(*HelloRequest)(nil),   // 0: example.helloworld.v1.HelloRequest
 	(*HelloResponse)(nil),  // 1: example.helloworld.v1.HelloResponse
 	(*GoodbyeRequest)(nil), // 2: example.helloworld.v1.GoodbyeRequest
@@ -254,7 +254,7 @@ func file_example_helloworld_v1_example_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_example_helloworld_v1_example_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_example_helloworld_v1_example_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*HelloRequest); i {
 			case 0:
 				return &v.state
@@ -266,7 +266,7 @@ func file_example_helloworld_v1_example_proto_init() {
 				return nil
 			}
 		}
-		file_example_helloworld_v1_example_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_example_helloworld_v1_example_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*HelloResponse); i {
 			case 0:
 				return &v.state
@@ -278,7 +278,7 @@ func file_example_helloworld_v1_example_proto_init() {
 				return nil
 			}
 		}
-		file_example_helloworld_v1_example_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_example_helloworld_v1_example_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*GoodbyeRequest); i {
 			case 0:
 				return &v.state
diff --git a/gen/example/helloworld/v1/example_temporal.pb.go b/gen/example/helloworld/v1/example_temporal.pb.go
index 52e867e5..80cedfe9 100644
--- a/gen/example/helloworld/v1/example_temporal.pb.go
+++ b/gen/example/helloworld/v1/example_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/helloworld/v1/example.proto
diff --git a/gen/example/helloworld/v1/helloworldv1xns/example_xns_temporal.pb.go b/gen/example/helloworld/v1/helloworldv1xns/example_xns_temporal.pb.go
index 72cf08ae..41ed1cef 100644
--- a/gen/example/helloworld/v1/helloworldv1xns/example_xns_temporal.pb.go
+++ b/gen/example/helloworld/v1/helloworldv1xns/example_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/helloworld/v1/example.proto
diff --git a/gen/example/mutex/v1/mutex.pb.go b/gen/example/mutex/v1/mutex.pb.go
index 64f58cc1..ed4a9957 100644
--- a/gen/example/mutex/v1/mutex.pb.go
+++ b/gen/example/mutex/v1/mutex.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: example/mutex/v1/mutex.proto
 
@@ -374,7 +374,7 @@ func file_example_mutex_v1_mutex_proto_rawDescGZIP() []byte {
 }
 
 var file_example_mutex_v1_mutex_proto_msgTypes = make([]protoimpl.MessageInfo, 5)
-var file_example_mutex_v1_mutex_proto_goTypes = []interface{}{
+var file_example_mutex_v1_mutex_proto_goTypes = []any{
 	(*MutexInput)(nil),                   // 0: example.mutex.v1.MutexInput
 	(*AcquireLockInput)(nil),             // 1: example.mutex.v1.AcquireLockInput
 	(*LockAcquiredInput)(nil),            // 2: example.mutex.v1.LockAcquiredInput
@@ -409,7 +409,7 @@ func file_example_mutex_v1_mutex_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_example_mutex_v1_mutex_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_example_mutex_v1_mutex_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*MutexInput); i {
 			case 0:
 				return &v.state
@@ -421,7 +421,7 @@ func file_example_mutex_v1_mutex_proto_init() {
 				return nil
 			}
 		}
-		file_example_mutex_v1_mutex_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_example_mutex_v1_mutex_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*AcquireLockInput); i {
 			case 0:
 				return &v.state
@@ -433,7 +433,7 @@ func file_example_mutex_v1_mutex_proto_init() {
 				return nil
 			}
 		}
-		file_example_mutex_v1_mutex_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_example_mutex_v1_mutex_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*LockAcquiredInput); i {
 			case 0:
 				return &v.state
@@ -445,7 +445,7 @@ func file_example_mutex_v1_mutex_proto_init() {
 				return nil
 			}
 		}
-		file_example_mutex_v1_mutex_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
+		file_example_mutex_v1_mutex_proto_msgTypes[3].Exporter = func(v any, i int) any {
 			switch v := v.(*ReleaseLockInput); i {
 			case 0:
 				return &v.state
@@ -457,7 +457,7 @@ func file_example_mutex_v1_mutex_proto_init() {
 				return nil
 			}
 		}
-		file_example_mutex_v1_mutex_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
+		file_example_mutex_v1_mutex_proto_msgTypes[4].Exporter = func(v any, i int) any {
 			switch v := v.(*SampleWorkflowWithMutexInput); i {
 			case 0:
 				return &v.state
diff --git a/gen/example/mutex/v1/mutex_temporal.pb.go b/gen/example/mutex/v1/mutex_temporal.pb.go
index 7aba04a3..c9ddb2ff 100644
--- a/gen/example/mutex/v1/mutex_temporal.pb.go
+++ b/gen/example/mutex/v1/mutex_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/mutex/v1/mutex.proto
diff --git a/gen/example/mutex/v1/mutexv1xns/mutex_xns_temporal.pb.go b/gen/example/mutex/v1/mutexv1xns/mutex_xns_temporal.pb.go
index 830e6781..50543d78 100644
--- a/gen/example/mutex/v1/mutexv1xns/mutex_xns_temporal.pb.go
+++ b/gen/example/mutex/v1/mutexv1xns/mutex_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/mutex/v1/mutex.proto
diff --git a/gen/example/searchattributes/v1/searchattributes.pb.go b/gen/example/searchattributes/v1/searchattributes.pb.go
index 32e09930..693643ab 100644
--- a/gen/example/searchattributes/v1/searchattributes.pb.go
+++ b/gen/example/searchattributes/v1/searchattributes.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: example/searchattributes/v1/searchattributes.proto
 
@@ -207,7 +207,7 @@ func file_example_searchattributes_v1_searchattributes_proto_rawDescGZIP() []byt
 }
 
 var file_example_searchattributes_v1_searchattributes_proto_msgTypes = make([]protoimpl.MessageInfo, 1)
-var file_example_searchattributes_v1_searchattributes_proto_goTypes = []interface{}{
+var file_example_searchattributes_v1_searchattributes_proto_goTypes = []any{
 	(*SearchAttributesInput)(nil), // 0: example.searchattributes.v1.SearchAttributesInput
 	(*timestamppb.Timestamp)(nil), // 1: google.protobuf.Timestamp
 	(*emptypb.Empty)(nil),         // 2: google.protobuf.Empty
@@ -229,7 +229,7 @@ func file_example_searchattributes_v1_searchattributes_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_example_searchattributes_v1_searchattributes_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_example_searchattributes_v1_searchattributes_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*SearchAttributesInput); i {
 			case 0:
 				return &v.state
diff --git a/gen/example/searchattributes/v1/searchattributes_temporal.pb.go b/gen/example/searchattributes/v1/searchattributes_temporal.pb.go
index 132d89ae..660e3585 100644
--- a/gen/example/searchattributes/v1/searchattributes_temporal.pb.go
+++ b/gen/example/searchattributes/v1/searchattributes_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/searchattributes/v1/searchattributes.proto
diff --git a/gen/example/searchattributes/v1/searchattributesv1xns/searchattributes_xns_temporal.pb.go b/gen/example/searchattributes/v1/searchattributesv1xns/searchattributes_xns_temporal.pb.go
index 08ae8542..50c9ec65 100644
--- a/gen/example/searchattributes/v1/searchattributesv1xns/searchattributes_xns_temporal.pb.go
+++ b/gen/example/searchattributes/v1/searchattributesv1xns/searchattributes_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/searchattributes/v1/searchattributes.proto
diff --git a/gen/example/updatabletimer/v1/updatabletimer.pb.go b/gen/example/updatabletimer/v1/updatabletimer.pb.go
index 560f9ca7..903f870b 100644
--- a/gen/example/updatabletimer/v1/updatabletimer.pb.go
+++ b/gen/example/updatabletimer/v1/updatabletimer.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: example/updatabletimer/v1/updatabletimer.proto
 
@@ -266,7 +266,7 @@ func file_example_updatabletimer_v1_updatabletimer_proto_rawDescGZIP() []byte {
 }
 
 var file_example_updatabletimer_v1_updatabletimer_proto_msgTypes = make([]protoimpl.MessageInfo, 3)
-var file_example_updatabletimer_v1_updatabletimer_proto_goTypes = []interface{}{
+var file_example_updatabletimer_v1_updatabletimer_proto_goTypes = []any{
 	(*GetWakeUpTimeOutput)(nil),   // 0: example.updatabletimer.v1.GetWakeUpTimeOutput
 	(*UpdatableTimerInput)(nil),   // 1: example.updatabletimer.v1.UpdatableTimerInput
 	(*UpdateWakeUpTimeInput)(nil), // 2: example.updatabletimer.v1.UpdateWakeUpTimeInput
@@ -296,7 +296,7 @@ func file_example_updatabletimer_v1_updatabletimer_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_example_updatabletimer_v1_updatabletimer_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_example_updatabletimer_v1_updatabletimer_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*GetWakeUpTimeOutput); i {
 			case 0:
 				return &v.state
@@ -308,7 +308,7 @@ func file_example_updatabletimer_v1_updatabletimer_proto_init() {
 				return nil
 			}
 		}
-		file_example_updatabletimer_v1_updatabletimer_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_example_updatabletimer_v1_updatabletimer_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*UpdatableTimerInput); i {
 			case 0:
 				return &v.state
@@ -320,7 +320,7 @@ func file_example_updatabletimer_v1_updatabletimer_proto_init() {
 				return nil
 			}
 		}
-		file_example_updatabletimer_v1_updatabletimer_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_example_updatabletimer_v1_updatabletimer_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*UpdateWakeUpTimeInput); i {
 			case 0:
 				return &v.state
diff --git a/gen/example/updatabletimer/v1/updatabletimer_temporal.pb.go b/gen/example/updatabletimer/v1/updatabletimer_temporal.pb.go
index 2e1b18d1..528ced84 100644
--- a/gen/example/updatabletimer/v1/updatabletimer_temporal.pb.go
+++ b/gen/example/updatabletimer/v1/updatabletimer_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/updatabletimer/v1/updatabletimer.proto
diff --git a/gen/example/updatabletimer/v1/updatabletimerv1xns/updatabletimer_xns_temporal.pb.go b/gen/example/updatabletimer/v1/updatabletimerv1xns/updatabletimer_xns_temporal.pb.go
index c21a36e2..41f8f686 100644
--- a/gen/example/updatabletimer/v1/updatabletimerv1xns/updatabletimer_xns_temporal.pb.go
+++ b/gen/example/updatabletimer/v1/updatabletimerv1xns/updatabletimer_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/updatabletimer/v1/updatabletimer.proto
diff --git a/gen/example/v1/example.pb.go b/gen/example/v1/example.pb.go
index 130117bf..f0724a0f 100644
--- a/gen/example/v1/example.pb.go
+++ b/gen/example/v1/example.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: example/v1/example.proto
 
@@ -479,7 +479,7 @@ func file_example_v1_example_proto_rawDescGZIP() []byte {
 
 var file_example_v1_example_proto_enumTypes = make([]protoimpl.EnumInfo, 1)
 var file_example_v1_example_proto_msgTypes = make([]protoimpl.MessageInfo, 6)
-var file_example_v1_example_proto_goTypes = []interface{}{
+var file_example_v1_example_proto_goTypes = []any{
 	(Foo_Status)(0),                // 0: example.v1.Foo.Status
 	(*CreateFooRequest)(nil),       // 1: example.v1.CreateFooRequest
 	(*CreateFooResponse)(nil),      // 2: example.v1.CreateFooResponse
@@ -516,7 +516,7 @@ func file_example_v1_example_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_example_v1_example_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_example_v1_example_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*CreateFooRequest); i {
 			case 0:
 				return &v.state
@@ -528,7 +528,7 @@ func file_example_v1_example_proto_init() {
 				return nil
 			}
 		}
-		file_example_v1_example_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_example_v1_example_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*CreateFooResponse); i {
 			case 0:
 				return &v.state
@@ -540,7 +540,7 @@ func file_example_v1_example_proto_init() {
 				return nil
 			}
 		}
-		file_example_v1_example_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_example_v1_example_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*Foo); i {
 			case 0:
 				return &v.state
@@ -552,7 +552,7 @@ func file_example_v1_example_proto_init() {
 				return nil
 			}
 		}
-		file_example_v1_example_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
+		file_example_v1_example_proto_msgTypes[3].Exporter = func(v any, i int) any {
 			switch v := v.(*GetFooProgressResponse); i {
 			case 0:
 				return &v.state
@@ -564,7 +564,7 @@ func file_example_v1_example_proto_init() {
 				return nil
 			}
 		}
-		file_example_v1_example_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
+		file_example_v1_example_proto_msgTypes[4].Exporter = func(v any, i int) any {
 			switch v := v.(*NotifyRequest); i {
 			case 0:
 				return &v.state
@@ -576,7 +576,7 @@ func file_example_v1_example_proto_init() {
 				return nil
 			}
 		}
-		file_example_v1_example_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
+		file_example_v1_example_proto_msgTypes[5].Exporter = func(v any, i int) any {
 			switch v := v.(*SetFooProgressRequest); i {
 			case 0:
 				return &v.state
diff --git a/gen/example/v1/example_temporal.pb.go b/gen/example/v1/example_temporal.pb.go
index db45e80f..5e203fbe 100644
--- a/gen/example/v1/example_temporal.pb.go
+++ b/gen/example/v1/example_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/v1/example.proto
@@ -639,6 +639,8 @@ func (o *UpdateFooProgressOptions) Build(workflowID string, runID string, req *S
 	// set WaitPolicy
 	if v := o.waitPolicy; v != client.WorkflowUpdateStageUnspecified {
 		opts.WaitForStage = v
+	} else if opts.WaitForStage == client.WorkflowUpdateStageUnspecified {
+		opts.WaitForStage = client.WorkflowUpdateStageCompleted
 	}
 	return opts, nil
 }
diff --git a/gen/example/v1/examplev1xns/example_xns_temporal.pb.go b/gen/example/v1/examplev1xns/example_xns_temporal.pb.go
index 4c81f566..aca0229a 100644
--- a/gen/example/v1/examplev1xns/example_xns_temporal.pb.go
+++ b/gen/example/v1/examplev1xns/example_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/v1/example.proto
diff --git a/gen/example/xns/v1/xns.pb.go b/gen/example/xns/v1/xns.pb.go
index c48d43d8..6eba1fd2 100644
--- a/gen/example/xns/v1/xns.pb.go
+++ b/gen/example/xns/v1/xns.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: example/xns/v1/xns.proto
 
@@ -600,7 +600,7 @@ func file_example_xns_v1_xns_proto_rawDescGZIP() []byte {
 
 var file_example_xns_v1_xns_proto_enumTypes = make([]protoimpl.EnumInfo, 1)
 var file_example_xns_v1_xns_proto_msgTypes = make([]protoimpl.MessageInfo, 8)
-var file_example_xns_v1_xns_proto_goTypes = []interface{}{
+var file_example_xns_v1_xns_proto_goTypes = []any{
 	(Foo_Status)(0),                // 0: example.xns.v1.Foo.Status
 	(*CreateFooRequest)(nil),       // 1: example.xns.v1.CreateFooRequest
 	(*CreateFooResponse)(nil),      // 2: example.xns.v1.CreateFooResponse
@@ -642,7 +642,7 @@ func file_example_xns_v1_xns_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_example_xns_v1_xns_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*CreateFooRequest); i {
 			case 0:
 				return &v.state
@@ -654,7 +654,7 @@ func file_example_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_example_xns_v1_xns_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*CreateFooResponse); i {
 			case 0:
 				return &v.state
@@ -666,7 +666,7 @@ func file_example_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_example_xns_v1_xns_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*Foo); i {
 			case 0:
 				return &v.state
@@ -678,7 +678,7 @@ func file_example_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_example_xns_v1_xns_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[3].Exporter = func(v any, i int) any {
 			switch v := v.(*GetFooProgressResponse); i {
 			case 0:
 				return &v.state
@@ -690,7 +690,7 @@ func file_example_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_example_xns_v1_xns_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[4].Exporter = func(v any, i int) any {
 			switch v := v.(*NotifyRequest); i {
 			case 0:
 				return &v.state
@@ -702,7 +702,7 @@ func file_example_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_example_xns_v1_xns_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[5].Exporter = func(v any, i int) any {
 			switch v := v.(*ProvisionFooRequest); i {
 			case 0:
 				return &v.state
@@ -714,7 +714,7 @@ func file_example_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_example_xns_v1_xns_proto_msgTypes[6].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[6].Exporter = func(v any, i int) any {
 			switch v := v.(*ProvisionFooResponse); i {
 			case 0:
 				return &v.state
@@ -726,7 +726,7 @@ func file_example_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_example_xns_v1_xns_proto_msgTypes[7].Exporter = func(v interface{}, i int) interface{} {
+		file_example_xns_v1_xns_proto_msgTypes[7].Exporter = func(v any, i int) any {
 			switch v := v.(*SetFooProgressRequest); i {
 			case 0:
 				return &v.state
diff --git a/gen/example/xns/v1/xns_temporal.pb.go b/gen/example/xns/v1/xns_temporal.pb.go
index 377c2de9..4d312f92 100644
--- a/gen/example/xns/v1/xns_temporal.pb.go
+++ b/gen/example/xns/v1/xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/xns/v1/xns.proto
@@ -1586,6 +1586,8 @@ func (o *UpdateFooProgressOptions) Build(workflowID string, runID string, req *S
 	// set WaitPolicy
 	if v := o.waitPolicy; v != client.WorkflowUpdateStageUnspecified {
 		opts.WaitForStage = v
+	} else if opts.WaitForStage == client.WorkflowUpdateStageUnspecified {
+		opts.WaitForStage = client.WorkflowUpdateStageCompleted
 	}
 	return opts, nil
 }
diff --git a/gen/example/xns/v1/xnsv1xns/xns_xns_temporal.pb.go b/gen/example/xns/v1/xnsv1xns/xns_xns_temporal.pb.go
index 5765d4e9..35751aa7 100644
--- a/gen/example/xns/v1/xnsv1xns/xns_xns_temporal.pb.go
+++ b/gen/example/xns/v1/xnsv1xns/xns_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: example/xns/v1/xns.proto
diff --git a/gen/patch/go.pb.go b/gen/patch/go.pb.go
index 5e9d367f..5b8129ea 100644
--- a/gen/patch/go.pb.go
+++ b/gen/patch/go.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: patch/go.proto
 
@@ -412,7 +412,7 @@ func file_patch_go_proto_rawDescGZIP() []byte {
 }
 
 var file_patch_go_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
-var file_patch_go_proto_goTypes = []interface{}{
+var file_patch_go_proto_goTypes = []any{
 	(*Options)(nil),                       // 0: go.Options
 	(*LintOptions)(nil),                   // 1: go.LintOptions
 	(*descriptorpb.MessageOptions)(nil),   // 2: google.protobuf.MessageOptions
@@ -448,7 +448,7 @@ func file_patch_go_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_patch_go_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_patch_go_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*Options); i {
 			case 0:
 				return &v.state
@@ -460,7 +460,7 @@ func file_patch_go_proto_init() {
 				return nil
 			}
 		}
-		file_patch_go_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_patch_go_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*LintOptions); i {
 			case 0:
 				return &v.state
diff --git a/gen/temporal/v1/temporal.pb.go b/gen/temporal/v1/temporal.pb.go
index b36b4a5b..dce212ae 100644
--- a/gen/temporal/v1/temporal.pb.go
+++ b/gen/temporal/v1/temporal.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: temporal/v1/temporal.proto
 
@@ -1965,7 +1965,7 @@ func file_temporal_v1_temporal_proto_rawDescGZIP() []byte {
 
 var file_temporal_v1_temporal_proto_enumTypes = make([]protoimpl.EnumInfo, 6)
 var file_temporal_v1_temporal_proto_msgTypes = make([]protoimpl.MessageInfo, 14)
-var file_temporal_v1_temporal_proto_goTypes = []interface{}{
+var file_temporal_v1_temporal_proto_goTypes = []any{
 	(CLIFeature)(0),                     // 0: temporal.v1.CLIFeature
 	(IDReusePolicy)(0),                  // 1: temporal.v1.IDReusePolicy
 	(ParentClosePolicy)(0),              // 2: temporal.v1.ParentClosePolicy
@@ -2059,7 +2059,7 @@ func file_temporal_v1_temporal_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_temporal_v1_temporal_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*ActivityOptions); i {
 			case 0:
 				return &v.state
@@ -2071,7 +2071,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*CLIOptions); i {
 			case 0:
 				return &v.state
@@ -2083,7 +2083,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*CommandOptions); i {
 			case 0:
 				return &v.state
@@ -2095,7 +2095,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[3].Exporter = func(v any, i int) any {
 			switch v := v.(*Patch); i {
 			case 0:
 				return &v.state
@@ -2107,7 +2107,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[4].Exporter = func(v any, i int) any {
 			switch v := v.(*QueryOptions); i {
 			case 0:
 				return &v.state
@@ -2119,7 +2119,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[5].Exporter = func(v any, i int) any {
 			switch v := v.(*RetryPolicy); i {
 			case 0:
 				return &v.state
@@ -2131,7 +2131,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[6].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[6].Exporter = func(v any, i int) any {
 			switch v := v.(*ServiceOptions); i {
 			case 0:
 				return &v.state
@@ -2143,7 +2143,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[7].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[7].Exporter = func(v any, i int) any {
 			switch v := v.(*SignalOptions); i {
 			case 0:
 				return &v.state
@@ -2155,7 +2155,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[8].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[8].Exporter = func(v any, i int) any {
 			switch v := v.(*UpdateOptions); i {
 			case 0:
 				return &v.state
@@ -2167,7 +2167,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[9].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[9].Exporter = func(v any, i int) any {
 			switch v := v.(*WorkflowOptions); i {
 			case 0:
 				return &v.state
@@ -2179,7 +2179,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[10].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[10].Exporter = func(v any, i int) any {
 			switch v := v.(*XNSActivityOptions); i {
 			case 0:
 				return &v.state
@@ -2191,7 +2191,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[11].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[11].Exporter = func(v any, i int) any {
 			switch v := v.(*WorkflowOptions_Query); i {
 			case 0:
 				return &v.state
@@ -2203,7 +2203,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[12].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[12].Exporter = func(v any, i int) any {
 			switch v := v.(*WorkflowOptions_Signal); i {
 			case 0:
 				return &v.state
@@ -2215,7 +2215,7 @@ func file_temporal_v1_temporal_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_v1_temporal_proto_msgTypes[13].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_v1_temporal_proto_msgTypes[13].Exporter = func(v any, i int) any {
 			switch v := v.(*WorkflowOptions_Update); i {
 			case 0:
 				return &v.state
diff --git a/gen/temporal/xns/v1/xns.pb.go b/gen/temporal/xns/v1/xns.pb.go
index 73f78f48..0675aab2 100644
--- a/gen/temporal/xns/v1/xns.pb.go
+++ b/gen/temporal/xns/v1/xns.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: temporal/xns/v1/xns.proto
 
@@ -1020,7 +1020,7 @@ func file_temporal_xns_v1_xns_proto_rawDescGZIP() []byte {
 
 var file_temporal_xns_v1_xns_proto_enumTypes = make([]protoimpl.EnumInfo, 3)
 var file_temporal_xns_v1_xns_proto_msgTypes = make([]protoimpl.MessageInfo, 7)
-var file_temporal_xns_v1_xns_proto_goTypes = []interface{}{
+var file_temporal_xns_v1_xns_proto_goTypes = []any{
 	(IDReusePolicy)(0),                       // 0: temporal.xns.v1.IDReusePolicy
 	(ParentClosePolicy)(0),                   // 1: temporal.xns.v1.ParentClosePolicy
 	(WaitPolicy)(0),                          // 2: temporal.xns.v1.WaitPolicy
@@ -1074,7 +1074,7 @@ func file_temporal_xns_v1_xns_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_temporal_xns_v1_xns_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_xns_v1_xns_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*QueryRequest); i {
 			case 0:
 				return &v.state
@@ -1086,7 +1086,7 @@ func file_temporal_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_xns_v1_xns_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_xns_v1_xns_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*RetryPolicy); i {
 			case 0:
 				return &v.state
@@ -1098,7 +1098,7 @@ func file_temporal_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_xns_v1_xns_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_xns_v1_xns_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*SignalRequest); i {
 			case 0:
 				return &v.state
@@ -1110,7 +1110,7 @@ func file_temporal_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_xns_v1_xns_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_xns_v1_xns_proto_msgTypes[3].Exporter = func(v any, i int) any {
 			switch v := v.(*StartWorkflowOptions); i {
 			case 0:
 				return &v.state
@@ -1122,7 +1122,7 @@ func file_temporal_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_xns_v1_xns_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_xns_v1_xns_proto_msgTypes[4].Exporter = func(v any, i int) any {
 			switch v := v.(*UpdateRequest); i {
 			case 0:
 				return &v.state
@@ -1134,7 +1134,7 @@ func file_temporal_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_xns_v1_xns_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_xns_v1_xns_proto_msgTypes[5].Exporter = func(v any, i int) any {
 			switch v := v.(*UpdateWorkflowWithOptionsRequest); i {
 			case 0:
 				return &v.state
@@ -1146,7 +1146,7 @@ func file_temporal_xns_v1_xns_proto_init() {
 				return nil
 			}
 		}
-		file_temporal_xns_v1_xns_proto_msgTypes[6].Exporter = func(v interface{}, i int) interface{} {
+		file_temporal_xns_v1_xns_proto_msgTypes[6].Exporter = func(v any, i int) any {
 			switch v := v.(*WorkflowRequest); i {
 			case 0:
 				return &v.state
diff --git a/gen/test/expression/v1/expression.pb.go b/gen/test/expression/v1/expression.pb.go
index 40d6c642..4c433f1a 100644
--- a/gen/test/expression/v1/expression.pb.go
+++ b/gen/test/expression/v1/expression.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: test/expression/v1/expression.proto
 
@@ -316,7 +316,7 @@ func file_test_expression_v1_expression_proto_rawDescGZIP() []byte {
 }
 
 var file_test_expression_v1_expression_proto_msgTypes = make([]protoimpl.MessageInfo, 3)
-var file_test_expression_v1_expression_proto_goTypes = []interface{}{
+var file_test_expression_v1_expression_proto_goTypes = []any{
 	(*Request)(nil),                         // 0: temporal.v1.test.expression.v1.Request
 	(*Request_OuterNested)(nil),             // 1: temporal.v1.test.expression.v1.Request.OuterNested
 	(*Request_OuterNested_InnerNested)(nil), // 2: temporal.v1.test.expression.v1.Request.OuterNested.InnerNested
@@ -339,7 +339,7 @@ func file_test_expression_v1_expression_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_test_expression_v1_expression_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_test_expression_v1_expression_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*Request); i {
 			case 0:
 				return &v.state
@@ -351,7 +351,7 @@ func file_test_expression_v1_expression_proto_init() {
 				return nil
 			}
 		}
-		file_test_expression_v1_expression_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_test_expression_v1_expression_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*Request_OuterNested); i {
 			case 0:
 				return &v.state
@@ -363,7 +363,7 @@ func file_test_expression_v1_expression_proto_init() {
 				return nil
 			}
 		}
-		file_test_expression_v1_expression_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_test_expression_v1_expression_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*Request_OuterNested_InnerNested); i {
 			case 0:
 				return &v.state
diff --git a/gen/test/option/v1/option.pb.go b/gen/test/option/v1/option.pb.go
index 7807814b..400daf6e 100644
--- a/gen/test/option/v1/option.pb.go
+++ b/gen/test/option/v1/option.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: test/option/v1/option.proto
 
@@ -295,7 +295,7 @@ func file_test_option_v1_option_proto_rawDescGZIP() []byte {
 }
 
 var file_test_option_v1_option_proto_msgTypes = make([]protoimpl.MessageInfo, 4)
-var file_test_option_v1_option_proto_goTypes = []interface{}{
+var file_test_option_v1_option_proto_goTypes = []any{
 	(*ActivityWithInputRequest)(nil),  // 0: test.option.v1.ActivityWithInputRequest
 	(*ActivityWithInputResponse)(nil), // 1: test.option.v1.ActivityWithInputResponse
 	(*UpdateWithInputRequest)(nil),    // 2: test.option.v1.UpdateWithInputRequest
@@ -322,7 +322,7 @@ func file_test_option_v1_option_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_test_option_v1_option_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_test_option_v1_option_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*ActivityWithInputRequest); i {
 			case 0:
 				return &v.state
@@ -334,7 +334,7 @@ func file_test_option_v1_option_proto_init() {
 				return nil
 			}
 		}
-		file_test_option_v1_option_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_test_option_v1_option_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*ActivityWithInputResponse); i {
 			case 0:
 				return &v.state
@@ -346,7 +346,7 @@ func file_test_option_v1_option_proto_init() {
 				return nil
 			}
 		}
-		file_test_option_v1_option_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_test_option_v1_option_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*UpdateWithInputRequest); i {
 			case 0:
 				return &v.state
@@ -358,7 +358,7 @@ func file_test_option_v1_option_proto_init() {
 				return nil
 			}
 		}
-		file_test_option_v1_option_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
+		file_test_option_v1_option_proto_msgTypes[3].Exporter = func(v any, i int) any {
 			switch v := v.(*WorkflowWithInputRequest); i {
 			case 0:
 				return &v.state
diff --git a/gen/test/option/v1/option_temporal.pb.go b/gen/test/option/v1/option_temporal.pb.go
index 45a1fc11..c9f4836a 100644
--- a/gen/test/option/v1/option_temporal.pb.go
+++ b/gen/test/option/v1/option_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: test/option/v1/option.proto
diff --git a/gen/test/option/v1/optionv1xns/option_xns_temporal.pb.go b/gen/test/option/v1/optionv1xns/option_xns_temporal.pb.go
index 54e58927..aae7f082 100644
--- a/gen/test/option/v1/optionv1xns/option_xns_temporal.pb.go
+++ b/gen/test/option/v1/optionv1xns/option_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: test/option/v1/option.proto
diff --git a/gen/test/simple/common/v1/common.pb.go b/gen/test/simple/common/v1/common.pb.go
index e631f50b..db9f32d0 100644
--- a/gen/test/simple/common/v1/common.pb.go
+++ b/gen/test/simple/common/v1/common.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: test/simple/common/v1/common.proto
 
@@ -184,7 +184,7 @@ func file_test_simple_common_v1_common_proto_rawDescGZIP() []byte {
 }
 
 var file_test_simple_common_v1_common_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
-var file_test_simple_common_v1_common_proto_goTypes = []interface{}{
+var file_test_simple_common_v1_common_proto_goTypes = []any{
 	(*PaginatedRequest)(nil),  // 0: mycompany.simple.common.v1.PaginatedRequest
 	(*PaginatedResponse)(nil), // 1: mycompany.simple.common.v1.PaginatedResponse
 	(*anypb.Any)(nil),         // 2: google.protobuf.Any
@@ -204,7 +204,7 @@ func file_test_simple_common_v1_common_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_test_simple_common_v1_common_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_common_v1_common_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*PaginatedRequest); i {
 			case 0:
 				return &v.state
@@ -216,7 +216,7 @@ func file_test_simple_common_v1_common_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_common_v1_common_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_common_v1_common_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*PaginatedResponse); i {
 			case 0:
 				return &v.state
diff --git a/gen/test/simple/v1/simple.pb.go b/gen/test/simple/v1/simple.pb.go
index 680db418..47bc77e0 100644
--- a/gen/test/simple/v1/simple.pb.go
+++ b/gen/test/simple/v1/simple.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: test/simple/v1/simple.proto
 
@@ -2134,7 +2134,7 @@ func file_test_simple_v1_simple_proto_rawDescGZIP() []byte {
 
 var file_test_simple_v1_simple_proto_enumTypes = make([]protoimpl.EnumInfo, 1)
 var file_test_simple_v1_simple_proto_msgTypes = make([]protoimpl.MessageInfo, 28)
-var file_test_simple_v1_simple_proto_goTypes = []interface{}{
+var file_test_simple_v1_simple_proto_goTypes = []any{
 	(OtherEnum)(0),                   // 0: mycompany.simple.OtherEnum
 	(*SomeWorkflow1Request)(nil),     // 1: mycompany.simple.SomeWorkflow1Request
 	(*SomeWorkflow1Response)(nil),    // 2: mycompany.simple.SomeWorkflow1Response
@@ -2254,7 +2254,7 @@ func file_test_simple_v1_simple_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_test_simple_v1_simple_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeWorkflow1Request); i {
 			case 0:
 				return &v.state
@@ -2266,7 +2266,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeWorkflow1Response); i {
 			case 0:
 				return &v.state
@@ -2278,7 +2278,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeWorkflow3Request); i {
 			case 0:
 				return &v.state
@@ -2290,7 +2290,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[3].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeActivity2Request); i {
 			case 0:
 				return &v.state
@@ -2302,7 +2302,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[4].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeActivity3Request); i {
 			case 0:
 				return &v.state
@@ -2314,7 +2314,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[5].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeActivity3Response); i {
 			case 0:
 				return &v.state
@@ -2326,7 +2326,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[6].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[6].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeQuery1Response); i {
 			case 0:
 				return &v.state
@@ -2338,7 +2338,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[7].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[7].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeQuery2Request); i {
 			case 0:
 				return &v.state
@@ -2350,7 +2350,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[8].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[8].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeQuery2Response); i {
 			case 0:
 				return &v.state
@@ -2362,7 +2362,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[9].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[9].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeSignal2Request); i {
 			case 0:
 				return &v.state
@@ -2374,7 +2374,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[10].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[10].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeSignal3Request); i {
 			case 0:
 				return &v.state
@@ -2386,7 +2386,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[11].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[11].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeSignal3Response); i {
 			case 0:
 				return &v.state
@@ -2398,7 +2398,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[12].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[12].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeUpdate1Request); i {
 			case 0:
 				return &v.state
@@ -2410,7 +2410,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[13].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[13].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeUpdate1Response); i {
 			case 0:
 				return &v.state
@@ -2422,7 +2422,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[14].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[14].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherWorkflowRequest); i {
 			case 0:
 				return &v.state
@@ -2434,7 +2434,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[15].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[15].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherWorkflowResponse); i {
 			case 0:
 				return &v.state
@@ -2446,7 +2446,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[16].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[16].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherQueryResponse); i {
 			case 0:
 				return &v.state
@@ -2458,7 +2458,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[17].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[17].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherSignalRequest); i {
 			case 0:
 				return &v.state
@@ -2470,7 +2470,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[18].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[18].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherUpdateRequest); i {
 			case 0:
 				return &v.state
@@ -2482,7 +2482,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[19].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[19].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherUpdateResponse); i {
 			case 0:
 				return &v.state
@@ -2494,7 +2494,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[20].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[20].Exporter = func(v any, i int) any {
 			switch v := v.(*Foo); i {
 			case 0:
 				return &v.state
@@ -2506,7 +2506,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[21].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[21].Exporter = func(v any, i int) any {
 			switch v := v.(*Qux); i {
 			case 0:
 				return &v.state
@@ -2518,7 +2518,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[22].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[22].Exporter = func(v any, i int) any {
 			switch v := v.(*WhatRequest); i {
 			case 0:
 				return &v.state
@@ -2530,7 +2530,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[23].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[23].Exporter = func(v any, i int) any {
 			switch v := v.(*LonelyActivity1Request); i {
 			case 0:
 				return &v.state
@@ -2542,7 +2542,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[24].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[24].Exporter = func(v any, i int) any {
 			switch v := v.(*LonelyActivity1Response); i {
 			case 0:
 				return &v.state
@@ -2554,7 +2554,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[25].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[25].Exporter = func(v any, i int) any {
 			switch v := v.(*SomeDeprecatedMessage); i {
 			case 0:
 				return &v.state
@@ -2566,7 +2566,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[26].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[26].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherWorkflowRequest_Bar); i {
 			case 0:
 				return &v.state
@@ -2578,7 +2578,7 @@ func file_test_simple_v1_simple_proto_init() {
 				return nil
 			}
 		}
-		file_test_simple_v1_simple_proto_msgTypes[27].Exporter = func(v interface{}, i int) interface{} {
+		file_test_simple_v1_simple_proto_msgTypes[27].Exporter = func(v any, i int) any {
 			switch v := v.(*OtherWorkflowRequest_Baz); i {
 			case 0:
 				return &v.state
@@ -2591,7 +2591,7 @@ func file_test_simple_v1_simple_proto_init() {
 			}
 		}
 	}
-	file_test_simple_v1_simple_proto_msgTypes[14].OneofWrappers = []interface{}{
+	file_test_simple_v1_simple_proto_msgTypes[14].OneofWrappers = []any{
 		(*OtherWorkflowRequest_Foo)(nil),
 		(*OtherWorkflowRequest_Bar_)(nil),
 		(*OtherWorkflowRequest_ExampleBytes)(nil),
diff --git a/gen/test/simple/v1/simple_temporal.pb.go b/gen/test/simple/v1/simple_temporal.pb.go
index 09161341..74b1658e 100644
--- a/gen/test/simple/v1/simple_temporal.pb.go
+++ b/gen/test/simple/v1/simple_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: test/simple/v1/simple.proto
@@ -6867,6 +6867,8 @@ func (o *OtherUpdateOptions) Build(workflowID string, runID string, req *OtherUp
 	// set WaitPolicy
 	if v := o.waitPolicy; v != client.WorkflowUpdateStageUnspecified {
 		opts.WaitForStage = v
+	} else if opts.WaitForStage == client.WorkflowUpdateStageUnspecified {
+		opts.WaitForStage = client.WorkflowUpdateStageCompleted
 	}
 	return opts, nil
 }
@@ -10615,6 +10617,8 @@ func (o *SomeDeprecatedUpdate1Options) Build(workflowID string, runID string, re
 	// set WaitPolicy
 	if v := o.waitPolicy; v != client.WorkflowUpdateStageUnspecified {
 		opts.WaitForStage = v
+	} else if opts.WaitForStage == client.WorkflowUpdateStageUnspecified {
+		opts.WaitForStage = client.WorkflowUpdateStageCompleted
 	}
 	return opts, nil
 }
@@ -10735,6 +10739,8 @@ func (o *SomeDeprecatedUpdate2Options) Build(workflowID string, runID string, re
 	// set WaitPolicy
 	if v := o.waitPolicy; v != client.WorkflowUpdateStageUnspecified {
 		opts.WaitForStage = v
+	} else if opts.WaitForStage == client.WorkflowUpdateStageUnspecified {
+		opts.WaitForStage = client.WorkflowUpdateStageCompleted
 	}
 	return opts, nil
 }
diff --git a/gen/test/simple/v1/v1xns/simple_xns_temporal.pb.go b/gen/test/simple/v1/v1xns/simple_xns_temporal.pb.go
index 0d8736f8..1167b83c 100644
--- a/gen/test/simple/v1/v1xns/simple_xns_temporal.pb.go
+++ b/gen/test/simple/v1/v1xns/simple_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: test/simple/v1/simple.proto
diff --git a/gen/test/xnserr/v1/xnserr.pb.go b/gen/test/xnserr/v1/xnserr.pb.go
index 039a67a8..0b83d4e5 100644
--- a/gen/test/xnserr/v1/xnserr.pb.go
+++ b/gen/test/xnserr/v1/xnserr.pb.go
@@ -1,6 +1,6 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.34.1
+// 	protoc-gen-go v1.34.2
 // 	protoc        (unknown)
 // source: test/xnserr/v1/xnserr.proto
 
@@ -394,7 +394,7 @@ func file_test_xnserr_v1_xnserr_proto_rawDescGZIP() []byte {
 
 var file_test_xnserr_v1_xnserr_proto_enumTypes = make([]protoimpl.EnumInfo, 1)
 var file_test_xnserr_v1_xnserr_proto_msgTypes = make([]protoimpl.MessageInfo, 3)
-var file_test_xnserr_v1_xnserr_proto_goTypes = []interface{}{
+var file_test_xnserr_v1_xnserr_proto_goTypes = []any{
 	(FailureInfo)(0),                // 0: test.xnserr.v1.FailureInfo
 	(*CallSleepRequest)(nil),        // 1: test.xnserr.v1.CallSleepRequest
 	(*SleepRequest)(nil),            // 2: test.xnserr.v1.SleepRequest
@@ -429,7 +429,7 @@ func file_test_xnserr_v1_xnserr_proto_init() {
 		return
 	}
 	if !protoimpl.UnsafeEnabled {
-		file_test_xnserr_v1_xnserr_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+		file_test_xnserr_v1_xnserr_proto_msgTypes[0].Exporter = func(v any, i int) any {
 			switch v := v.(*CallSleepRequest); i {
 			case 0:
 				return &v.state
@@ -441,7 +441,7 @@ func file_test_xnserr_v1_xnserr_proto_init() {
 				return nil
 			}
 		}
-		file_test_xnserr_v1_xnserr_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+		file_test_xnserr_v1_xnserr_proto_msgTypes[1].Exporter = func(v any, i int) any {
 			switch v := v.(*SleepRequest); i {
 			case 0:
 				return &v.state
@@ -453,7 +453,7 @@ func file_test_xnserr_v1_xnserr_proto_init() {
 				return nil
 			}
 		}
-		file_test_xnserr_v1_xnserr_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+		file_test_xnserr_v1_xnserr_proto_msgTypes[2].Exporter = func(v any, i int) any {
 			switch v := v.(*Failure); i {
 			case 0:
 				return &v.state
diff --git a/gen/test/xnserr/v1/xnserr_temporal.pb.go b/gen/test/xnserr/v1/xnserr_temporal.pb.go
index 63ae2412..e228425b 100644
--- a/gen/test/xnserr/v1/xnserr_temporal.pb.go
+++ b/gen/test/xnserr/v1/xnserr_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: test/xnserr/v1/xnserr.proto
diff --git a/gen/test/xnserr/v1/xnserrv1xns/xnserr_xns_temporal.pb.go b/gen/test/xnserr/v1/xnserrv1xns/xnserr_xns_temporal.pb.go
index 2cba2dc8..8a0226a9 100644
--- a/gen/test/xnserr/v1/xnserrv1xns/xnserr_xns_temporal.pb.go
+++ b/gen/test/xnserr/v1/xnserrv1xns/xnserr_xns_temporal.pb.go
@@ -1,8 +1,8 @@
 // Code generated by protoc-gen-go_temporal. DO NOT EDIT.
 // versions:
 //
-//	protoc-gen-go_temporal 1.13.4-next (5d73dd32b5c1ed2313641d4ed7326289e2d00e18)
-//	go go1.22.4
+//	protoc-gen-go_temporal 1.14.1-next (f7565e015d371e688f27fb0d8ad4c7747764beae)
+//	go go1.22.2
 //	protoc (unknown)
 //
 // source: test/xnserr/v1/xnserr.proto
diff --git a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_CreateFooRun.go b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_CreateFooRun.go
index 57f4cd76..e8b48111 100644
--- a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_CreateFooRun.go
+++ b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_CreateFooRun.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package examplev1
 
diff --git a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_ExampleClient.go b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_ExampleClient.go
index 97468527..ca94b1ac 100644
--- a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_ExampleClient.go
+++ b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_ExampleClient.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package examplev1
 
diff --git a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_UpdateFooProgressHandle.go b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_UpdateFooProgressHandle.go
index 0be8c44f..aa874a75 100644
--- a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_UpdateFooProgressHandle.go
+++ b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/example/v1/mock_UpdateFooProgressHandle.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package examplev1
 
diff --git a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/simple/v1/mock_SimpleWorkflowFunctions.go b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/simple/v1/mock_SimpleWorkflowFunctions.go
index 2193714f..33af9d9a 100644
--- a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/simple/v1/mock_SimpleWorkflowFunctions.go
+++ b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/simple/v1/mock_SimpleWorkflowFunctions.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package v1
 
diff --git a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_ServerClient.go b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_ServerClient.go
index ea2e12e8..276e2ff4 100644
--- a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_ServerClient.go
+++ b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_ServerClient.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package xnserrv1mocks
 
diff --git a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_SleepRun.go b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_SleepRun.go
index 6e167b4b..9d0d5313 100644
--- a/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_SleepRun.go
+++ b/mocks/github.com/cludden/protoc-gen-go-temporal/gen/test/xnserr/v1/mock_SleepRun.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package xnserrv1mocks
 
diff --git a/mocks/go.temporal.io/sdk/client/mock_Client.go b/mocks/go.temporal.io/sdk/client/mock_Client.go
index 7a5b7215..fa0424c1 100644
--- a/mocks/go.temporal.io/sdk/client/mock_Client.go
+++ b/mocks/go.temporal.io/sdk/client/mock_Client.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package client
 
diff --git a/mocks/go.temporal.io/sdk/clientutils/mock_WorkflowRun.go b/mocks/go.temporal.io/sdk/clientutils/mock_WorkflowRun.go
index 23a7ece9..e74ebcb5 100644
--- a/mocks/go.temporal.io/sdk/clientutils/mock_WorkflowRun.go
+++ b/mocks/go.temporal.io/sdk/clientutils/mock_WorkflowRun.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.43.0. DO NOT EDIT.
+// Code generated by mockery v2.43.2. DO NOT EDIT.
 
 package clientutils