diff --git a/cpp_tests/host_context/async_value_test.cc b/cpp_tests/host_context/async_value_test.cc
index 51f7aaf12d8..a837cea4c0c 100644
--- a/cpp_tests/host_context/async_value_test.cc
+++ b/cpp_tests/host_context/async_value_test.cc
@@ -104,7 +104,7 @@ TEST(AsyncValueTest, AddAndDropRef) {
 TEST(AsyncValueTest, KeepPayloadOnError) {
   int payload_value = 0;
 
-  struct Payload : tsl::AsyncPayload::KeepOnError {
+  struct Payload : internal::KeepAsyncValuePayloadOnError {
     explicit Payload(int* value) : value{value} { *value = 1; }
     ~Payload() { *value = 2; }
 
diff --git a/include/tfrt/host_context/async_value.h b/include/tfrt/host_context/async_value.h
index 24e19b44d4e..b03f145a404 100644
--- a/include/tfrt/host_context/async_value.h
+++ b/include/tfrt/host_context/async_value.h
@@ -32,7 +32,8 @@ namespace tfrt {
 using ::tsl::AsyncValue;  // NOLINT
 
 namespace internal {
-using ::tsl::internal::ConcreteAsyncValue;  // NOLINT
+using ::tsl::internal::ConcreteAsyncValue;            // NOLINT
+using ::tsl::internal::KeepAsyncValuePayloadOnError;  // NOLINT
 }  // namespace internal
 
 using ::tsl::DummyValueForErrorAsyncValue;  // NOLINT