diff --git a/tensorflow/compiler/xla/pjrt/pjrt_client.h b/tensorflow/compiler/xla/pjrt/pjrt_client.h
index 935f667142e..1bed959e3e6 100644
--- a/tensorflow/compiler/xla/pjrt/pjrt_client.h
+++ b/tensorflow/compiler/xla/pjrt/pjrt_client.h
@@ -695,7 +695,7 @@ struct ExecuteOptions {
   int32 launch_id = 0;
   // If non-null, an opaque context passed to an execution that may be used to
   // supply additional arguments to a derived class of PjRtExecutable.
-  std::unique_ptr<ExecuteContext> context;
+  ExecuteContext* context = nullptr;
 };
 
 // Represents a compiled computation that can be executed given handles to