diff --git a/xla/hlo/translate/hlo_to_mhlo/custom_call_importer.cc b/xla/hlo/translate/hlo_to_mhlo/custom_call_importer.cc
index 988375ab5cd4b..10a78e149ec97 100644
--- a/xla/hlo/translate/hlo_to_mhlo/custom_call_importer.cc
+++ b/xla/hlo/translate/hlo_to_mhlo/custom_call_importer.cc
@@ -16,6 +16,7 @@ limitations under the License.
 #include "xla/hlo/translate/hlo_to_mhlo/custom_call_importer.h"
 
 #include <cstdint>
+#include <string>
 #include <vector>
 
 #include "absl/status/statusor.h"
diff --git a/xla/hlo/translate/hlo_to_mhlo/hlo_utils_test.cc b/xla/hlo/translate/hlo_to_mhlo/hlo_utils_test.cc
index 28edbc3aab4c7..e51b0b9b325c2 100644
--- a/xla/hlo/translate/hlo_to_mhlo/hlo_utils_test.cc
+++ b/xla/hlo/translate/hlo_to_mhlo/hlo_utils_test.cc
@@ -16,7 +16,6 @@ limitations under the License.
 #include "xla/hlo/translate/hlo_to_mhlo/hlo_utils.h"
 
 #include <cstdint>
-#include <cstring>
 #include <vector>
 
 #include "mlir/IR/Builders.h"
diff --git a/xla/hlo/translate/hlo_to_mhlo/translate.cc b/xla/hlo/translate/hlo_to_mhlo/translate.cc
index 8fef24fead917..8bc72e6afaeff 100644
--- a/xla/hlo/translate/hlo_to_mhlo/translate.cc
+++ b/xla/hlo/translate/hlo_to_mhlo/translate.cc
@@ -14,6 +14,7 @@ limitations under the License.
 
 #include "xla/hlo/translate/hlo_to_mhlo/translate.h"
 
+#include <string>
 #include <utility>
 
 #include "absl/log/log.h"