diff --git a/tsl/platform/base64.cc b/tsl/platform/base64.cc
index f344f69d5..808f44522 100644
--- a/tsl/platform/base64.cc
+++ b/tsl/platform/base64.cc
@@ -18,6 +18,7 @@ limitations under the License.
 #include <cstdint>
 #include <cstring>
 #include <memory>
+#include <string>
 
 #include "absl/status/status.h"
 #include "absl/strings/string_view.h"
diff --git a/tsl/platform/unbounded_work_queue_test.cc b/tsl/platform/unbounded_work_queue_test.cc
index ea24a27f3..82a4c10b3 100644
--- a/tsl/platform/unbounded_work_queue_test.cc
+++ b/tsl/platform/unbounded_work_queue_test.cc
@@ -18,7 +18,6 @@ limitations under the License.
 #include <functional>
 #include <memory>
 
-#include "absl/memory/memory.h"
 #include "absl/synchronization/mutex.h"
 #include "xla/tsl/platform/env.h"
 #include "xla/tsl/platform/test.h"