diff --git a/workflows/workflow.go b/workflows/workflow.go
index 20ddcdf..db3c7e0 100644
--- a/workflows/workflow.go
+++ b/workflows/workflow.go
@@ -14,7 +14,7 @@ func TestRunWorkflow(ctx workflow.Context, input *pb.TestRunInput) (*pb.TestRunO
 	logger := workflow.GetLogger(ctx)
 	logger.Info("TestRunWorkflow started", "runID", input.RunId)
 
-	ao := workflow.ActivityOptions{
+	ctx = workflow.WithActivityOptions(ctx, workflow.ActivityOptions{
 		StartToCloseTimeout: 10 * time.Minute, // Activity 执行超时时间
 		HeartbeatTimeout:    30 * time.Second, // Heartbeat 防止 Worker 假死
 		RetryPolicy: &temporal.RetryPolicy{ // Activity 级别的重试策略
@@ -24,8 +24,7 @@ func TestRunWorkflow(ctx workflow.Context, input *pb.TestRunInput) (*pb.TestRunO
 			MaximumAttempts:        3,
 			NonRetryableErrorTypes: []string{"NonRetryableErrorType"}, // 自定义不可重试的错误
 		},
-	}
-	ctx = workflow.WithActivityOptions(ctx, ao)
+	})
 
 	var (
 		apiResults        []*pb.ApiTestResult